fix code, Manually download the model

This commit is contained in:
billwuhao
2025-04-26 18:03:28 +08:00
parent 33a9b1a8a3
commit 09e43b6e76
2 changed files with 5 additions and 3 deletions
+4 -2
View File
@@ -164,6 +164,7 @@ class DiffRhythmRun:
self.cfm = None
self.vae = None
self.muq = None
self.tokenizer = None
@classmethod
def INPUT_TYPES(cls):
@@ -213,9 +214,9 @@ class DiffRhythmRun:
max_frames = 6144
if self.cfm is None:
self.cfm, tokenizer, self.muq, self.vae = self.prepare_model(model, self.device)
self.cfm, self.tokenizer, self.muq, self.vae = self.prepare_model(model, self.device)
lrc_prompt, start_time = get_lrc_token(max_frames, lyrics_prompt, tokenizer, self.device)
lrc_prompt, start_time = get_lrc_token(max_frames, lyrics_prompt, self.tokenizer, self.device)
vocal_flag = False
if style_audio:
@@ -256,6 +257,7 @@ class DiffRhythmRun:
self.cfm = None
self.muq = None
self.vae = None
self.tokenizer = None
gc.collect()
torch.cuda.empty_cache()
+1 -1
View File
@@ -1,7 +1,7 @@
[project]
name = "diffrhythm_mw"
description = "Blazingly Fast and Embarrassingly Simple End-to-End Full-Length Song Generation. A node for ComfyUI."
version = "2.1.4"
version = "2.1.5"
license = {file = "LICENSE"}
dependencies = ["# accelerate==1.4.0", "# torchdiffeq==0.2.5", "# torchaudio==2.6.0", "# transformers==4.49.0", "# librosa==0.10.2.post1", "# pyarrow==19.0.1", "# pandas==2.2.3", "# bitsandbytes", "# jieba==0.42.1", "# cn2an==0.5.23", "# pypinyin==0.53.0", "# onnxruntime", "LangSegment", "x-transformers", "pylance", "ema-pytorch", "prefigure", "muq", "mutagen", "pyopenjtalk", "pykakasi", "Unidecode", "phonemizer"]