rahul7star commited on
Commit
89f1ae7
·
verified ·
1 Parent(s): 5433ca6

Update simple_app.py

Browse files
Files changed (1) hide show
  1. simple_app.py +3 -0
simple_app.py CHANGED
@@ -47,6 +47,8 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
47
  "--sample_shift", "8",
48
  "--sample_guide_scale", "6",
49
  "--prompt", prompt,
 
 
50
  "--save_file", "generated_video.mp4"
51
  ]
52
  print("Starting video generation process...")
@@ -58,6 +60,7 @@ def infer(prompt, progress=gr.Progress(track_tqdm=True)):
58
  bufsize=1)
59
 
60
  print("Process started successfully.")
 
61
 
62
  while True:
63
  # Poll stdout with a 40ms timeout.
 
47
  "--sample_shift", "8",
48
  "--sample_guide_scale", "6",
49
  "--prompt", prompt,
50
+ "--t5_cpu",
51
+ "--offload_model", True ,
52
  "--save_file", "generated_video.mp4"
53
  ]
54
  print("Starting video generation process...")
 
60
  bufsize=1)
61
 
62
  print("Process started successfully.")
63
+ print(process)
64
 
65
  while True:
66
  # Poll stdout with a 40ms timeout.