akhaliq HF staff commited on
Commit
dbd0ad6
1 Parent(s): 57885e4

set device to cuda

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ args = edict({
97
  "upsample_type": "nearest",
98
  "num_frequencies": 6,
99
  "deform_face_scale_coef": 0.0,
100
- "device": "cpu"
101
  })
102
 
103
  # download FLAME and DECA pretrained
 
97
  "upsample_type": "nearest",
98
  "num_frequencies": 6,
99
  "deform_face_scale_coef": 0.0,
100
+ "device": "cuda"
101
  })
102
 
103
  # download FLAME and DECA pretrained