Hugging Face's logo

Spaces:
Statical
/
Image
Running on Zero

App Files Community
Staticaliza commited on
Commit
f54b0fd
·
verified ·
1 Parent(s): 2feb008

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ DEFAULT_WIDTH = 1024
32
 
33
  headers = {"Content-Type": "application/json", "Authorization": f"Bearer {HF_TOKEN}" }
34
 
35
- repo_large_path = snapshot_download(repo_id="stabilityai/stable-diffusion-3-medium", revision="refs/pr/26", token=HF_TOKEN)
36
  repo_large_transformer_path = SD3Transformer2DModel.from_pretrained(repo_large_path, torch_dtype=torch.float16)
37
  repo_large_transformer = PeftModel.from_pretrained(repo_large_transformer_path, "jasperai/flash-sd3")
38
 
 
32
 
33
  headers = {"Content-Type": "application/json", "Authorization": f"Bearer {HF_TOKEN}" }
34
 
35
+ repo_large_path = snapshot_download(repo_id="stabilityai/stable-diffusion-3-medium", revision="refs/pr/26", repo_type="model", token=HF_TOKEN)
36
  repo_large_transformer_path = SD3Transformer2DModel.from_pretrained(repo_large_path, torch_dtype=torch.float16)
37
  repo_large_transformer = PeftModel.from_pretrained(repo_large_transformer_path, "jasperai/flash-sd3")
38