Spaces:

VideoSys
/

CogVideoX

Runtime error

VictorKai1996NUS commited on Aug 28, 2024

Commit

d6e8791

verified ·

1 Parent(s): b71d548

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,6 +11,8 @@ import logging
 import gradio as gr
 from videosys import CogVideoConfig, VideoSysEngine
 from videosys.models.cogvideo.pipeline import CogVideoPABConfig
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)

 import gradio as gr
 from videosys import CogVideoConfig, VideoSysEngine
 from videosys.models.cogvideo.pipeline import CogVideoPABConfig
+import psutil
+import GPUtil
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)