Spaces:
Runtime error
Runtime error
update rumway stable diffusion to realistic vision model
Browse files
app.py
CHANGED
|
@@ -25,7 +25,7 @@ controlnet = ControlNetModel.from_pretrained(
|
|
| 25 |
torch_dtype=torch.float16,
|
| 26 |
)
|
| 27 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
| 28 |
-
"
|
| 29 |
).to("cuda")
|
| 30 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
| 31 |
|
|
|
|
| 25 |
torch_dtype=torch.float16,
|
| 26 |
)
|
| 27 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
| 28 |
+
"stablediffusionapi/realistic-vision-v51", controlnet=controlnet, torch_dtype=torch.float16
|
| 29 |
).to("cuda")
|
| 30 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
| 31 |
|