vilarin commited on
Commit
6663f0e
·
verified ·
1 Parent(s): b18804c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -78,7 +78,7 @@ def generate(
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
 
78
  max_cfg_scale=2,
79
  contrast_enhance_scale=1.2
80
  )
81
+ model_manager.to("cpu")
82
  export_to_video(frames, video_path, fps=fps_id)
83
 
84
  return video_path, seed