Spaces:

ysmao
/

Layout-Control

Running on Zero

ysmao commited on Sep 29

Commit

237c7d9

•

1 Parent(s): 103dcbe

update rumway stable diffusion to realistic vision model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ controlnet = ControlNetModel.from_pretrained(
     torch_dtype=torch.float16,
 )
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
 ).to("cuda")
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)

     torch_dtype=torch.float16,
 )
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
+    "stablediffusionapi/realistic-vision-v51", controlnet=controlnet, torch_dtype=torch.float16
 ).to("cuda")
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)