Skip to content

Commit 5a2bc0d

Browse files
committed
Added a default case in load checkpoint
1 parent 9256066 commit 5a2bc0d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

ai_diffusion/workflow.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,10 @@ def load_checkpoint_with_lora(w: ComfyWorkflow, checkpoint: CheckpointInput, mod
105105
model = w.nunchaku_load_flux_diffusion_model(model_info.filename, cache_threshold=cache)
106106
elif model_info.arch in (Arch.qwen, Arch.qwen_e):
107107
model = w.nunchaku_load_qwen_diffusion_model(model_info.filename, cpu_offload="enable", num_blocks_on_gpu=16, use_pin_memory="disable")
108+
else:
109+
raise RuntimeError(
110+
f"Style checkpoint {checkpoint.checkpoint} has an unsupported quantized format {model_info.format.name}"
111+
)
108112
case _:
109113
raise RuntimeError(
110114
f"Style checkpoint {checkpoint.checkpoint} has an unsupported format {model_info.format.name}"

0 commit comments

Comments
 (0)