Spaces:

Pie31415
/

rome

Runtime error

akhaliq HF staff commited on Jan 1, 2023

Commit

320db74

•

1 Parent(s): 57885e4

set device to cuda (#2)

- set device to cuda (dbd0ad66f1ca2c19107e3ae5fe1a4bb0886782be)

Co-authored-by: Ahsen Khaliq <akhaliq@users.noreply.huggingface.co>

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