[Minor] fix deprecation warnings for torch_dtype
Browse files- configuration_rnd.py +1 -1
configuration_rnd.py
CHANGED
|
@@ -42,7 +42,7 @@ CONFIG_DEFAULTS = {
|
|
| 42 |
"router_aux_loss_coef": 0.001,
|
| 43 |
"sliding_window": False,
|
| 44 |
"tie_word_embeddings": False,
|
| 45 |
-
"
|
| 46 |
"use_cache": False,
|
| 47 |
"use_sliding_window": False,
|
| 48 |
"vocab_size": 151936,
|
|
|
|
| 42 |
"router_aux_loss_coef": 0.001,
|
| 43 |
"sliding_window": False,
|
| 44 |
"tie_word_embeddings": False,
|
| 45 |
+
"dtype": "bfloat16",
|
| 46 |
"use_cache": False,
|
| 47 |
"use_sliding_window": False,
|
| 48 |
"vocab_size": 151936,
|