Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
95d9369
1
Parent(s):
00f6a78
Update app.py
Browse files
app.py
CHANGED
@@ -17,13 +17,13 @@ BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
|
|
17 |
# Initialize both pipelines
|
18 |
vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
|
19 |
#init_pipe = DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V5.1_noVAE", torch_dtype=torch.float16)
|
20 |
-
controlnet = ControlNetModel.from_pretrained("monster-labs/control_v1p_sd15_qrcode_monster"
|
21 |
main_pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
22 |
BASE_MODEL,
|
23 |
controlnet=controlnet,
|
24 |
vae=vae,
|
25 |
safety_checker=None,
|
26 |
-
torch_dtype=torch.float16,
|
27 |
).to("cuda")
|
28 |
#model_id = "stabilityai/sd-x2-latent-upscaler"
|
29 |
#upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
|
|
17 |
# Initialize both pipelines
|
18 |
vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
|
19 |
#init_pipe = DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V5.1_noVAE", torch_dtype=torch.float16)
|
20 |
+
controlnet = ControlNetModel.from_pretrained("monster-labs/control_v1p_sd15_qrcode_monster")#, torch_dtype=torch.float16)
|
21 |
main_pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
22 |
BASE_MODEL,
|
23 |
controlnet=controlnet,
|
24 |
vae=vae,
|
25 |
safety_checker=None,
|
26 |
+
#torch_dtype=torch.float16,
|
27 |
).to("cuda")
|
28 |
#model_id = "stabilityai/sd-x2-latent-upscaler"
|
29 |
#upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|