wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-22 09:33:29 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

More small fixes

Browse Source
This commit is contained in:
kijai 2026-03-05 01:27:24 +02:00
parent c2aa8b0a25
commit 13613be86c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/sd.py
View File

@ -432,7 +432,7 @@ class CLIP:
self.load_model(tokens) self.load_model(tokens)
self.cond_stage_model.set_clip_options({"layer": None}) self.cond_stage_model.set_clip_options({"layer": None})
self.cond_stage_model.set_clip_options({"execution_device": self.patcher.load_device}) self.cond_stage_model.set_clip_options({"execution_device": self.patcher.load_device})
return self.cond_stage_model.generate(tokens, do_sample=do_sample, max_length=max_length, temperature=temperature, top_k=top_k, top_p=top_p, min_p=min_p, repetition_penalty=repetition_penalty, presence_penalty=presence_penalty, seed=seed) return self.cond_stage_model.generate(tokens, do_sample=do_sample, max_length=max_length, temperature=temperature, top_k=top_k, top_p=top_p, min_p=min_p, repetition_penalty=repetition_penalty, seed=seed, presence_penalty=presence_penalty)
def decode(self, token_ids, skip_special_tokens=True): def decode(self, token_ids, skip_special_tokens=True):
return self.tokenizer.decode(token_ids, skip_special_tokens=skip_special_tokens) return self.tokenizer.decode(token_ids, skip_special_tokens=skip_special_tokens)
@ -1250,7 +1250,7 @@ def detect_te_model(sd):
return TEModel.QWEN25_3B return TEModel.QWEN25_3B
if weight.shape[0] == 512: if weight.shape[0] == 512:
return TEModel.QWEN25_7B return TEModel.QWEN25_7B
if "model.language_model.layers.0.linear_attn.A_log" in sd: if "model.language_model.layers.0.linear_attn.A_log" in sd and "model.language_model.layers.0.input_layernorm.weight" in sd:
weight = sd['model.language_model.layers.0.input_layernorm.weight'] weight = sd['model.language_model.layers.0.input_layernorm.weight']
if weight.shape[0] == 1024: if weight.shape[0] == 1024:
return TEModel.QWEN35_08B return TEModel.QWEN35_08B