Spaces:

AIML-TUDA
/

semantic-diffusion

Sleeping

mbrack commited on Feb 13, 2023

Commit

394c8f2

•

1 Parent(s): 1bf488a

Fix SD checkpoint typo

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from diffusers import SemanticStableDiffusionPipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = SemanticStableDiffusionPipeline.from_pretrained("/runwayml/stable-diffusion-v1-5/")
 pipe = pipe.to(device)
 gen = torch.Generator(device=device)

 device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = SemanticStableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
 pipe = pipe.to(device)
 gen = torch.Generator(device=device)