SamarthPujari commited on
Commit
d9d5d2a
·
verified ·
1 Parent(s): 772aacf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ hf_token = os.getenv("HF_TOKEN")
20
  login(token=hf_token)
21
 
22
  sd_pipe = StableDiffusionPipeline.from_pretrained(
23
- "runwayml/stable-diffusion-v1-5",
24
  use_auth_token=hf_token,
25
  torch_dtype=torch.float16 if device == "cuda" else torch.float32
26
  ).to(device)
 
20
  login(token=hf_token)
21
 
22
  sd_pipe = StableDiffusionPipeline.from_pretrained(
23
+ "rupeshs/LCM-runwayml-stable-diffusion-v1-5",
24
  use_auth_token=hf_token,
25
  torch_dtype=torch.float16 if device == "cuda" else torch.float32
26
  ).to(device)