vittore commited on
Commit
f165e6e
·
1 Parent(s): b64f780

Initialize both pipelines

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -30,7 +30,17 @@ if torch.cuda.is_available():
30
  else:
31
  device='cpu'
32
 
33
-
 
 
 
 
 
 
 
 
 
 
34
 
35
  def greet(name):
36
  return "Hello " + name + "!!"
 
30
  else:
31
  device='cpu'
32
 
33
+ # Initialize both pipelines
34
+ vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
35
+ #init_pipe = DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V5.1_noVAE", torch_dtype=torch.float16)
36
+ controlnet = ControlNetModel.from_pretrained("monster-labs/control_v1p_sd15_qrcode_monster", torch_dtype=torch.float16)#, torch_dtype=torch.float16)
37
+ main_pipe = StableDiffusionControlNetPipeline.from_pretrained(
38
+ BASE_MODEL,
39
+ controlnet=controlnet,
40
+ vae=vae,
41
+ safety_checker=None,
42
+ torch_dtype=torch.float16,
43
+ ).to(device)
44
 
45
  def greet(name):
46
  return "Hello " + name + "!!"