Spaces:
Sleeping
Sleeping
Change load device to cuda
Browse files- config.ini +1 -1
config.ini
CHANGED
|
@@ -21,7 +21,7 @@ opt_13b = str:facebook/opt-13b
|
|
| 21 |
|
| 22 |
[models.params]
|
| 23 |
dtype = str:float32
|
| 24 |
-
load_device = str:
|
| 25 |
run_device = str:cuda
|
| 26 |
|
| 27 |
[encrypt.default]
|
|
|
|
| 21 |
|
| 22 |
[models.params]
|
| 23 |
dtype = str:float32
|
| 24 |
+
load_device = str:cuda
|
| 25 |
run_device = str:cuda
|
| 26 |
|
| 27 |
[encrypt.default]
|