Commit
·
679791d
1
Parent(s):
947a0fa
no sd pipeline
Browse files
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')
|