Spaces:

kevalfst
/

visionary-ai

Running

kevalfst commited on Apr 29

Commit

8f45be3

verified ·

1 Parent(s): 16588c5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # Load Stable Diffusion v1.5 from Hugging Face
 pipe = StableDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-v1-5",
     torch_dtype=torch.float16 if device == "cuda" else torch.float32,
     revision="fp16" if device == "cuda" else None,
     use_safetensors=True

 # Load Stable Diffusion v1.5 from Hugging Face
 pipe = StableDiffusionPipeline.from_pretrained(
+    "stable-diffusion-v1-5/stable-diffusion-v1-5",
     torch_dtype=torch.float16 if device == "cuda" else torch.float32,
     revision="fp16" if device == "cuda" else None,
     use_safetensors=True