multimodalart HF staff commited on
Commit
95d9369
1 Parent(s): 00f6a78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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", 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)
 
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)