Spaces:

Pie31415
/

rome

Build error

Pie31415 commited on Jan 6, 2023

Commit

f1461a5

2 Parent(s): 9887bdf 320db74

Merge branch 'main' of https://huggingface.co/spaces/Pie31415/rome into main

Files changed (1) hide show

app.py CHANGED Viewed

@@ -107,7 +107,7 @@ args = edict({
     "upsample_type": "nearest",
     "num_frequencies": 6,
     "deform_face_scale_coef": 0.0,
-    "device": "cpu"
 })
 # download FLAME and DECA pretrained

     "upsample_type": "nearest",
     "num_frequencies": 6,
     "deform_face_scale_coef": 0.0,
+    "device": "cuda"
 })
 # download FLAME and DECA pretrained