Upload folder using huggingface_hub

modeling_mistral.py

@@ -1121,6 +1121,7 @@ class MistralForCausalLM(MistralPreTrainedModel):
         output_hidden_states: Optional[bool] = None,
         return_dict: Optional[bool] = None,
         reduction: Optional[str] = "mean",
+        use_fastv: Optional[bool] = False,
     ) -> Union[Tuple, CausalLMOutputWithPast]:
         r"""
         Args:

registry.py

@@ -31,16 +31,16 @@ class Registry:
         def wrap(model_cls):
             from .base_model import BaseModel
 
-            assert issubclass(
-                model_cls, BaseModel
-            ), "All models must inherit BaseModel class"
+            # assert issubclass(
+            #     model_cls, BaseModel
+            # ), "All models must inherit BaseModel class"
 
-            if name in cls.mapping["model_name_mapping"]:
-                raise KeyError(
-                    "Name '{}' already registered for {}.".format(
-                        name, cls.mapping["model_name_mapping"][name]
-                    )
-                )
+            # if name in cls.mapping["model_name_mapping"]:
+            #     raise KeyError(
+            #         "Name '{}' already registered for {}.".format(
+            #             name, cls.mapping["model_name_mapping"][name]
+            #         )
+            #     )
             cls.mapping["model_name_mapping"][name] = model_cls
             return model_cls