stanrom commited on
Commit
30a0641
·
verified ·
1 Parent(s): f77165e

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -10,7 +10,7 @@
10
  },
11
  "bias": true,
12
  "bos_token_id": 1,
13
- "device": "cuda",
14
  "eos_token_id": 2,
15
  "hidden_act": "silu",
16
  "hidden_size": 4096,
 
10
  },
11
  "bias": true,
12
  "bos_token_id": 1,
13
+ "device": "cpu",
14
  "eos_token_id": 2,
15
  "hidden_act": "silu",
16
  "hidden_size": 4096,