Update configuration_skywork_moe.py
Browse files
configuration_skywork_moe.py
CHANGED
@@ -13,7 +13,7 @@ LLAMA_PRETRAINED_CONFIG_ARCHIVE_MAP = {}
|
|
13 |
|
14 |
class SkyworkMoeConfig(PretrainedConfig):
|
15 |
|
16 |
-
model_type = "
|
17 |
keys_to_ignore_at_inference = ["past_key_values"]
|
18 |
|
19 |
def __init__(
|
|
|
13 |
|
14 |
class SkyworkMoeConfig(PretrainedConfig):
|
15 |
|
16 |
+
model_type = "skywork_moe"
|
17 |
keys_to_ignore_at_inference = ["past_key_values"]
|
18 |
|
19 |
def __init__(
|