EvanTHU commited on
Commit
d5e8fc2
·
verified ·
1 Parent(s): 17c85e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -123,6 +123,10 @@ def generate_video_from_text(text, opt, pipeline):
123
  save_path = pjoin(save_dir, str(filename))
124
  os.makedirs(save_dir, exist_ok=True)
125
 
 
 
 
 
126
  start_time = time.perf_counter()
127
  gr.Info("Generating motion...", duration = 3)
128
  pred_motions, _ = pipeline.generate(texts, torch.LongTensor([int(x) for x in motion_lens]))
 
123
  save_path = pjoin(save_dir, str(filename))
124
  os.makedirs(save_dir, exist_ok=True)
125
 
126
+ print("xxxxxxx")
127
+ print(pipeline.device)
128
+ print("xxxxxxx")
129
+
130
  start_time = time.perf_counter()
131
  gr.Info("Generating motion...", duration = 3)
132
  pred_motions, _ = pipeline.generate(texts, torch.LongTensor([int(x) for x in motion_lens]))