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,
|