QinOwen
commited on
Commit
·
f016f7c
1
Parent(s):
f7eee3e
fix-bug
Browse files
VADER-VideoCrafter/scripts/main/train_t2v_lora.py
CHANGED
@@ -710,7 +710,7 @@ def run_training(args, model, **kwargs):
|
|
710 |
mixed_precision=args.mixed_precision,
|
711 |
project_dir=args.project_dir,
|
712 |
device_placement=True,
|
713 |
-
cpu=
|
714 |
)
|
715 |
output_dir = args.project_dir
|
716 |
|
|
|
710 |
mixed_precision=args.mixed_precision,
|
711 |
project_dir=args.project_dir,
|
712 |
device_placement=True,
|
713 |
+
cpu=False
|
714 |
)
|
715 |
output_dir = args.project_dir
|
716 |
|
app.py
CHANGED
@@ -22,7 +22,7 @@ examples = [
|
|
22 |
|
23 |
model = setup_model()
|
24 |
|
25 |
-
|
26 |
def gradio_main_fn(prompt, lora_model, lora_rank, seed, height, width, unconditional_guidance_scale, ddim_steps, ddim_eta,
|
27 |
frames, savefps):
|
28 |
global model
|
|
|
22 |
|
23 |
model = setup_model()
|
24 |
|
25 |
+
@spaces.GPU(duration=120)
|
26 |
def gradio_main_fn(prompt, lora_model, lora_rank, seed, height, width, unconditional_guidance_scale, ddim_steps, ddim_eta,
|
27 |
frames, savefps):
|
28 |
global model
|