Spaces:
Runtime error
Runtime error
Update shape.py
Browse files
shape.py
CHANGED
@@ -23,7 +23,7 @@ pipe_kwargs = {
|
|
23 |
"requires_safety_checker": False,
|
24 |
"torch_dtype": torch.float16 ,
|
25 |
}
|
26 |
-
|
27 |
"stabilityai/stable-diffusion-2-1-base",
|
28 |
**pipe_kwargs,
|
29 |
)
|
|
|
23 |
"requires_safety_checker": False,
|
24 |
"torch_dtype": torch.float16 ,
|
25 |
}
|
26 |
+
pipe = StableDiffusionPipeline.from_pretrained(
|
27 |
"stabilityai/stable-diffusion-2-1-base",
|
28 |
**pipe_kwargs,
|
29 |
)
|