vilarin commited on
Commit
b18804c
·
verified ·
1 Parent(s): 925bdcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -77,8 +77,8 @@ def generate(
77
  min_cfg_scale=2,
78
  max_cfg_scale=2,
79
  contrast_enhance_scale=1.2
80
- ).frames[0]
81
-
82
  export_to_video(frames, video_path, fps=fps_id)
83
 
84
  return video_path, seed
 
77
  min_cfg_scale=2,
78
  max_cfg_scale=2,
79
  contrast_enhance_scale=1.2
80
+ )
81
+ model_manager.to(0)
82
  export_to_video(frames, video_path, fps=fps_id)
83
 
84
  return video_path, seed