Spaces:

fabrioop
/

BlaxTOP

Sleeping

fabrioop commited on 11 days ago

Commit

29085f8

verified ·

1 Parent(s): 6c888bb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5",
     torch_dtype=torch.float16 if device=="cuda" else torch.float32
 )
 pipe = pipe.to(device)

 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5",
     torch_dtype=torch.float16 if device=="cuda" else torch.float32
 )
 pipe = pipe.to(device)