Spaces:

Pie31415
/

rome

Build error

akhaliq HF staff commited on Dec 30, 2022

Commit

dbd0ad6

•

1 Parent(s): 57885e4

set device to cuda

Files changed (1) hide show

app.py CHANGED Viewed

@@ -97,7 +97,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