Pie31415 akhaliq HF staff commited on
Commit
320db74
1 Parent(s): 57885e4

set device to cuda (#2)

Browse files

- set device to cuda (dbd0ad66f1ca2c19107e3ae5fe1a4bb0886782be)


Co-authored-by: Ahsen Khaliq <[email protected]>

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