Nguyen17 commited on
Commit
080701d
·
verified ·
1 Parent(s): abedffe

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +1 -1
config.yaml CHANGED
@@ -18,4 +18,4 @@ generation:
18
  do_sample: True
19
  top_p: 0.9
20
  environment:
21
- device: 'cuda'
 
18
  do_sample: True
19
  top_p: 0.9
20
  environment:
21
+ device: 'cpu'