Ahsen Khaliq commited on
Commit
8bbcb84
1 Parent(s): d3aec3b

Update configs/prediction/default.yaml

Browse files
Files changed (1) hide show
  1. configs/prediction/default.yaml +1 -1
configs/prediction/default.yaml CHANGED
@@ -10,5 +10,5 @@ dataset:
10
  img_suffix: .png
11
  pad_out_to_modulo: 8
12
 
13
- device: cuda
14
  out_key: inpainted
 
10
  img_suffix: .png
11
  pad_out_to_modulo: 8
12
 
13
+ device: cpu
14
  out_key: inpainted