chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

external lora

Browse Source
This commit is contained in:
nick 2024-10-24 22:46:31 +08:00
parent 0779136134
commit 7fdfba6b6e
1 changed files with 34 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
comfy-nodes/external_lora.py
View File

@ -64,32 +64,42 @@ class ComfyUIDeployExternalLora:
import os import os
import uuid import uuid
if lora_url and lora_url.startswith("http"): if lora_url:
if lora_save_name: if lora_url.startswith("http"):
existing_loras = folder_paths.get_filename_list("loras") if lora_save_name:
# Check if lora_save_name exists in the list existing_loras = folder_paths.get_filename_list("loras")
if lora_save_name in existing_loras: # Check if lora_save_name exists in the list
print(f"using lora: {lora_save_name}") if lora_save_name in existing_loras:
return (lora_save_name,) print(f"using lora: {lora_save_name}")
return (lora_save_name,)
else:
lora_save_name = str(uuid.uuid4()) + ".safetensors"
print(lora_save_name)
print(folder_paths.folder_names_and_paths["loras"][0][0])
destination_path = os.path.join(
folder_paths.folder_names_and_paths["loras"][0][0], lora_save_name
)
print(destination_path)
print(
"Downloading external lora - "
+ lora_url
+ " to "
+ destination_path
)
response = requests.get(
lora_url,
headers={"User-Agent": "Mozilla/5.0"},
allow_redirects=True,
)
with open(destination_path, "wb") as out_file:
out_file.write(response.content)
print(f"Ext Lora loading: {lora_url} to {lora_save_name}")
return (lora_save_name,)
else: else:
lora_save_name = str(uuid.uuid4()) + ".safetensors" print(f"Ext Lora loading: {lora_url}")
print(lora_save_name) return (lora_url,)
print(folder_paths.folder_names_and_paths["loras"][0][0])
destination_path = os.path.join(
folder_paths.folder_names_and_paths["loras"][0][0], lora_save_name
)
print(destination_path)
print("Downloading external lora - " + lora_url + " to " + destination_path)
response = requests.get(
lora_url,
headers={"User-Agent": "Mozilla/5.0"},
allow_redirects=True,
)
with open(destination_path, "wb") as out_file:
out_file.write(response.content)
return (lora_save_name,)
else: else:
print(f"using lora: {default_lora_name}") print(f"Ext Lora loading: {default_lora_name}")
return (default_lora_name,) return (default_lora_name,)