diff --git a/CSMNode.py b/CSMNode.py index 8925ce7..e025eb1 100644 --- a/CSMNode.py +++ b/CSMNode.py @@ -214,6 +214,7 @@ MODEL_CACHE = None class CSMDialogRun: def __init__(self): self.device = "cuda" if torch.cuda.is_available() else "cpu" + self.model_name = None @classmethod def INPUT_TYPES(s): @@ -282,7 +283,8 @@ class CSMDialogRun: ): global MODEL_CACHE - if MODEL_CACHE is None: + if MODEL_CACHE is None or self.model_name != model: + self.model_name = model csm_1b_path = os.path.join(models_dir, "TTS", "csm-1b") config_path = os.path.join(csm_1b_path, "config.json") diff --git a/pyproject.toml b/pyproject.toml index 4b6f5e7..c2913d7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "csm_mw" description = "ComfyUI node of Conversational Speech Model (CSM)." -version = "1.1.1" +version = "1.1.2" license = {file = "LICENSE"} [project.urls]