Hugging Face's logo

Spaces:
fun-research
/
TiTok
Running on Zero

App Files Community
1
yucornetto commited on
Commit
ed3d5ce
·
verified ·
1 Parent(s): 547b9cf

test with cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import argparse
11
  import demo_util
12
  import os
13
 
14
- device = "cuda"
15
 
16
  model2ckpt = {
17
  "TiTok-L-32": ("tokenizer_titok_l32.bin", "generator_titok_l32.bin"),
 
11
  import demo_util
12
  import os
13
 
14
+ device = "cpu"
15
 
16
  model2ckpt = {
17
  "TiTok-L-32": ("tokenizer_titok_l32.bin", "generator_titok_l32.bin"),