Spaces:

primecai
/

diffusion-self-distillation

Running on Zero

primecai commited on 5 days ago

Commit

007257f

1 Parent(s): 05bfdcf

loading bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,8 @@ CHECKPOINT = "primecai/dsd_model"
 def init_pipeline():
     global pipe
     transformer = FluxTransformer2DConditionalModel.from_pretrained(
-        os.path.join(CHECKPOINT, "transformer"),
         torch_dtype=torch.bfloat16,
         low_cpu_mem_usage=False,
         ignore_mismatched_sizes=True,

 def init_pipeline():
     global pipe
     transformer = FluxTransformer2DConditionalModel.from_pretrained(
+        CHECKPOINT,
+        subfolder="transformer",
         torch_dtype=torch.bfloat16,
         low_cpu_mem_usage=False,
         ignore_mismatched_sizes=True,