qwen2vl-flux

Runtime error

LPX55 commited on Apr 7

Commit

bbece9f

verified ·

1 Parent(s): c13f18c

Update mini.py

Files changed (1) hide show

mini.py CHANGED Viewed

@@ -103,7 +103,7 @@ scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(
 qwen2vl = Qwen2VLSimplifiedModel.from_pretrained(
     os.path.join(MODEL_CACHE_DIR, "qwen2-vl"),
     quantization_config=quant_config,
-).to(DTYPE).cpu()
 # Load connector and embedder
 connector = Qwen2Connector().to(DTYPE).cpu()

 qwen2vl = Qwen2VLSimplifiedModel.from_pretrained(
     os.path.join(MODEL_CACHE_DIR, "qwen2-vl"),
     quantization_config=quant_config,
+)
 # Load connector and embedder
 connector = Qwen2Connector().to(DTYPE).cpu()