Spaces:

VIDraft
/

Portrait-Animation

Running on Zero

xiaozhongji commited on Jan 16

Commit

6c402fc

1 Parent(s): 4fb7040

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,10 @@ huggingface-cli download stabilityai/stable-video-diffusion-img2vid-xt --local-d
 huggingface-cli download openai/whisper-tiny --local-dir checkpoints/whisper-tiny'
 os.system(cmd)
-pipe = Sonic(0)
 def get_md5(content):
     md5hash = hashlib.md5(content)

 huggingface-cli download openai/whisper-tiny --local-dir checkpoints/whisper-tiny'
 os.system(cmd)
+import torch
+print('torch.cuda.device_count()', torch.cuda.device_count())
+pipe = Sonic()
 def get_md5(content):
     md5hash = hashlib.md5(content)