williamberman commited on
Commit
679791d
·
1 Parent(s): 947a0fa

no sd pipeline

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -47,11 +47,9 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=20, s
47
 
48
  pipe.vae.to('cuda')
49
  pipe.unet.to('cuda')
50
- pipe.controlnet.to('cuda')
51
  output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
52
  pipe.vae.to('cpu')
53
  pipe.unet.to('cpu')
54
- pipe.controlnet.to('cpu')
55
 
56
  comparing_unet.to('cuda')
57
  comparing_vae.to('cuda')
 
47
 
48
  pipe.vae.to('cuda')
49
  pipe.unet.to('cuda')
 
50
  output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
51
  pipe.vae.to('cpu')
52
  pipe.unet.to('cpu')
 
53
 
54
  comparing_unet.to('cuda')
55
  comparing_vae.to('cuda')