Commit
•
364bcfc
1
Parent(s):
98f6fc3
Set initial device to cuda:0 for faster initial loading
Browse files- config.json +1 -1
config.json
CHANGED
|
@@ -33,7 +33,7 @@
|
|
| 33 |
"name": "kaiming_normal_",
|
| 34 |
"verbose": 0
|
| 35 |
},
|
| 36 |
-
"init_device": "
|
| 37 |
"learned_pos_emb": true,
|
| 38 |
"logit_scale": null,
|
| 39 |
"max_seq_len": 8192,
|
|
|
|
| 33 |
"name": "kaiming_normal_",
|
| 34 |
"verbose": 0
|
| 35 |
},
|
| 36 |
+
"init_device": "cuda:0",
|
| 37 |
"learned_pos_emb": true,
|
| 38 |
"logit_scale": null,
|
| 39 |
"max_seq_len": 8192,
|