AP123 commited on
Commit
1e08323
β€’
1 Parent(s): 04cb33c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -13,10 +13,12 @@ from diffusers import (
13
  EulerDiscreteScheduler,
14
  )
15
 
 
16
  controlnet = ControlNetModel.from_pretrained(
17
  "monster-labs/control_v1p_sd15_qrcode_monster", torch_dtype=torch.float16
18
  )
19
 
 
20
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
21
  "SG161222/Realistic_Vision_V2.0",
22
  controlnet=controlnet,
 
13
  EulerDiscreteScheduler,
14
  )
15
 
16
+ # Load controlnet model in float16 precision
17
  controlnet = ControlNetModel.from_pretrained(
18
  "monster-labs/control_v1p_sd15_qrcode_monster", torch_dtype=torch.float16
19
  )
20
 
21
+ # Load the pipeline in float16 precision
22
  pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
23
  "SG161222/Realistic_Vision_V2.0",
24
  controlnet=controlnet,