set device to cuda

#2
by akhaliq HF staff - opened
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