xiaozhongji commited on
Commit
375fd74
·
1 Parent(s): 2fc2052
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -12,8 +12,8 @@ huggingface-cli download stabilityai/stable-video-diffusion-img2vid-xt --local-d
12
  huggingface-cli download openai/whisper-tiny --local-dir checkpoints/whisper-tiny;'
13
  os.system(cmd)
14
 
15
- import torch
16
- print('torch.cuda.device_count()', torch.cuda.device_count())
17
 
18
  pipe = Sonic()
19
 
 
12
  huggingface-cli download openai/whisper-tiny --local-dir checkpoints/whisper-tiny;'
13
  os.system(cmd)
14
 
15
+ # import torch
16
+ # print('torch.cuda.device_count()', torch.cuda.device_count())
17
 
18
  pipe = Sonic()
19
 
requirements.txt CHANGED
@@ -7,7 +7,7 @@ torchvision==0.17.2
7
  transformers==4.43.2
8
  imageio==2.31.1
9
  imageio-ffmpeg==0.5.1
10
- gradio==3.50.0
11
  omegaconf==2.3.0
12
  tqdm==4.65.2
13
  librosa==0.10.2.post1
 
7
  transformers==4.43.2
8
  imageio==2.31.1
9
  imageio-ffmpeg==0.5.1
10
+ gradio==4.3.0
11
  omegaconf==2.3.0
12
  tqdm==4.65.2
13
  librosa==0.10.2.post1