Skip to content

GitLab

Unverified Commit 5fcee4a4 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

[Single File] Fix loading (#10349) 

update
parent 76e2727b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/loaders/single_file_utils.py
View file @ 5fcee4a4
...@@ -626,7 +626,7 @@ def infer_diffusers_model_type(checkpoint): ...@@ -626,7 +626,7 @@ def infer_diffusers_model_type(checkpoint):
elif any(key in checkpoint for key in CHECKPOINT_KEY_NAMES["mochi-1-preview"]): elif any(key in checkpoint for key in CHECKPOINT_KEY_NAMES["mochi-1-preview"]):
model_type = "mochi-1-preview" model_type = "mochi-1-preview"
if CHECKPOINT_KEY_NAMES["hunyuan-video"] in checkpoint: elif CHECKPOINT_KEY_NAMES["hunyuan-video"] in checkpoint:
model_type = "hunyuan-video" model_type = "hunyuan-video"
else: else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment