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 |
)
|