Text Generation
Transformers
PyTorch
skywork_moe
custom_code
zhao1iang commited on
Commit
ede88fd
1 Parent(s): 6f38345

Update configuration_skywork_moe.py

Browse files
Files changed (1) hide show
  1. configuration_skywork_moe.py +1 -1
configuration_skywork_moe.py CHANGED
@@ -13,7 +13,7 @@ LLAMA_PRETRAINED_CONFIG_ARCHIVE_MAP = {}
13
 
14
  class SkyworkMoeConfig(PretrainedConfig):
15
 
16
- model_type = "skywork"
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__(