Spaces:

primecai
/

diffusion-self-distillation

Running on Zero

primecai commited on Mar 5, 2025

Commit

04e1bd0

1 Parent(s): 68c264b

add token

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ def init_pipeline():
         use_auth_token=os.getenv("HF_TOKEN"),
     )
     pipe.load_lora_weights(
-        pretrained_model_name_or_path="primecai/dsd_model",
         weight_name="pytorch_lora_weights.safetensors",
         use_auth_token=os.getenv("HF_TOKEN"),
     )

         use_auth_token=os.getenv("HF_TOKEN"),
     )
     pipe.load_lora_weights(
+        CHECKPOINT,
         weight_name="pytorch_lora_weights.safetensors",
         use_auth_token=os.getenv("HF_TOKEN"),
     )