Beijia11 commited on
Commit
9bd5727
·
1 Parent(s): 4f3b39d

install triton

Browse files
Files changed (2) hide show
  1. app.py +0 -3
  2. requirements.txt +1 -1
app.py CHANGED
@@ -10,9 +10,6 @@ from moviepy.editor import VideoFileClip
10
  from diffusers.utils import load_image, load_video
11
  import spaces
12
 
13
- import torch._dynamo
14
- torch._dynamo.config.suppress_errors = True
15
-
16
  project_root = os.path.dirname(os.path.abspath(__file__))
17
  os.environ["GRADIO_TEMP_DIR"] = os.path.join(project_root, "tmp", "gradio")
18
  sys.path.append(project_root)
 
10
  from diffusers.utils import load_image, load_video
11
  import spaces
12
 
 
 
 
13
  project_root = os.path.dirname(os.path.abspath(__file__))
14
  os.environ["GRADIO_TEMP_DIR"] = os.path.join(project_root, "tmp", "gradio")
15
  sys.path.append(project_root)
requirements.txt CHANGED
@@ -10,7 +10,7 @@ scikit-image==0.22.0
10
  scikit-learn==1.4.1.post1
11
  cupy-cuda11x
12
  accelerate
13
- yt-dlp
14
  pandas
15
 
16
  # cogvideox
 
10
  scikit-learn==1.4.1.post1
11
  cupy-cuda11x
12
  accelerate
13
+ triton
14
  pandas
15
 
16
  # cogvideox