jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Refactor diffusers model convert code to be able to reuse it.

Browse Source
This commit is contained in:
comfyanonymous
2023-05-28 01:52:09 -04:00
parent f3ac938b4a
commit 0fc483dcfd
3 changed files with 113 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
nodes.py
View File

@@ -17,7 +17,7 @@ import safetensors.torch
sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), "comfy"))
import comfy.diffusers_convert
import comfy.diffusers_load
import comfy.samplers
import comfy.sample
import comfy.sd
@@ -377,7 +377,7 @@ class DiffusersLoader:
model_path = path
break
return comfy.diffusers_convert.load_diffusers(model_path, fp16=comfy.model_management.should_use_fp16(), output_vae=output_vae, output_clip=output_clip, embedding_directory=folder_paths.get_folder_paths("embeddings"))
return comfy.diffusers_load.load_diffusers(model_path, fp16=comfy.model_management.should_use_fp16(), output_vae=output_vae, output_clip=output_clip, embedding_directory=folder_paths.get_folder_paths("embeddings"))
class unCLIPCheckpointLoader: