Marmik commited on
Commit
dd28f4f
·
verified ·
1 Parent(s): 336dc50

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_tiny_mixtral.py +3 -1
modeling_tiny_mixtral.py CHANGED
@@ -85,7 +85,9 @@ class TinyMixtralConfig(PretrainedConfig):
85
  "AutoConfig": "modeling_tiny_mixtral.TinyMixtralConfig",
86
  "AutoModelForCausalLM": "modeling_tiny_mixtral.TinyMixtralForCausalLM"
87
  }
88
- super().__init__(top_k=None,**kwargs)
 
 
89
  self.vocab_size = vocab_size
90
  self.d_model = d_model
91
  self.d_head = d_head
 
85
  "AutoConfig": "modeling_tiny_mixtral.TinyMixtralConfig",
86
  "AutoModelForCausalLM": "modeling_tiny_mixtral.TinyMixtralForCausalLM"
87
  }
88
+ # Remove top_k from kwargs if it exists to avoid conflict
89
+ # kwargs.pop('top_k', None)
90
+ super().__init__(**kwargs)
91
  self.vocab_size = vocab_size
92
  self.d_model = d_model
93
  self.d_head = d_head