nodes/pkpkTech-ComfyUI-TemporaryLoader
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Changed ckpt_type specification

Browse Source
This commit is contained in:
pkpk
2024-02-04 12:35:58 +09:00
parent 5329f570bf
commit aff56bab39
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
LoadTempCheckpoint.py
+3 -2
View File
@@ -18,7 +18,7 @@ class LoadTempCheckpoint:
return {
"required": {
"ckpt_url": ("STRING", {"default": ""}),
"ckpt_type": (["safetensors", "other"], {"default": "safetensors"}),
"ckpt_type": (["auto", "safetensors", "other"], {"default": "auto"}),
"download_split": ("INT", {"default": 4, "min": 1, "max": 8, "step": 1})
},
}
@@ -34,7 +34,8 @@ class LoadTempCheckpoint:
if bin is None:
raise file_name if file_name is not None else Exception("Download failed.")
sd = load_torch_bin(bin, ckpt_type=="safetensors" or file_name.endswith(".safetensors"))
is_safetensors = file_name.endswith(".safetensors") if ckpt_type =="auto" else ckpt_type == "safetensors"
sd = load_torch_bin(bin, is_safetensors)
sd_keys = sd.keys()
clip = None
clipvision = None
LoadTempLoRA.py
+3 -2
View File
@@ -19,7 +19,7 @@ class LoadTempLoRA:
"model": ("MODEL",),
"clip": ("CLIP", ),
"ckpt_url": ("STRING", {"default": ""}),
"ckpt_type": (["safetensors", "other"], {"default": "safetensors"}),
"ckpt_type": (["auto", "safetensors", "other"], {"default": "auto"}),
"download_split": ("INT", {"default": 4, "min": 1, "max": 8, "step": 1}),
"strength_model": ("FLOAT", {"default": 1.0, "min": -20.0, "max": 20.0, "step": 0.01}),
"strength_clip": ("FLOAT", {"default": 1.0, "min": -20.0, "max": 20.0, "step": 0.01}),
@@ -48,7 +48,8 @@ class LoadTempLoRA:
if bin is None:
raise file_name if file_name is not None else Exception("Download failed.")
lora = load_torch_bin(bin, ckpt_type=="safetensors" or file_name.endswith(".safetensors"), safe_load=True)
is_safetensors = file_name.endswith(".safetensors") if ckpt_type =="auto" else ckpt_type == "safetensors"
lora = load_torch_bin(bin, is_safetensors, safe_load=True)
self.loaded_lora = (ckpt_url, lora)
model_lora, clip_lora = comfy.sd.load_lora_for_models(model, clip, lora, strength_model, strength_clip)