Spaces:
Runtime error
Runtime error
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)
|