weathon
commited on
Commit
·
42ea091
1
Parent(s):
3ce4524
cuda
Browse files
app.py
CHANGED
|
@@ -66,8 +66,8 @@ def save_image(prompt: str, negative_prompt: str, img_vsf: Image, img_nag: Image
|
|
| 66 |
f.write("\n")
|
| 67 |
|
| 68 |
|
| 69 |
-
|
| 70 |
-
|
| 71 |
import os
|
| 72 |
@spaces.GPU
|
| 73 |
def generate_video(positive_prompt, negative_prompt, guidance_scale, bias, step, seed, nag_guidance, nag_alpha, nag_tau, nag_step, progress=gr.Progress(track_tqdm=False)):
|
|
|
|
| 66 |
f.write("\n")
|
| 67 |
|
| 68 |
|
| 69 |
+
pipe = pipe.to("cuda")
|
| 70 |
+
nag_pipe = nag_pipe.to("cuda")
|
| 71 |
import os
|
| 72 |
@spaces.GPU
|
| 73 |
def generate_video(positive_prompt, negative_prompt, guidance_scale, bias, step, seed, nag_guidance, nag_alpha, nag_tau, nag_step, progress=gr.Progress(track_tqdm=False)):
|