Spaces:

zhangyang-0123
/

EcoDiff-SD-XL

Sleeping

zhangyang-0123 commited on Dec 7, 2024

Commit

a014349

1 Parent(s): 78ea680

fix app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,6 @@ pruned_pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch
 pruned_pipe.transformer = torch.load(
     hf_hub_download("zhangyang-0123/EcoDiffPrunedModels", "model/sdxl/sdxl.pkl"),
     map_location="cpu",
-    torch_dtype=torch_dtype,
 )
 pruned_pipe = pruned_pipe.to(device)

 pruned_pipe.transformer = torch.load(
     hf_hub_download("zhangyang-0123/EcoDiffPrunedModels", "model/sdxl/sdxl.pkl"),
     map_location="cpu",
 )
 pruned_pipe = pruned_pipe.to(device)