Upload folder using huggingface_hub

configuration_helix_mrna.py

@@ -16,6 +16,7 @@ class HelixmRNAConfig(PretrainedConfig):
         Number of processes to use for data processing.
     """
 
+    model_type = "mamba2" # helical's model type is "mamba2" so this needs to be set
     model_name: Literal["helical-ai/Helix-mRNA"] = "helical-ai/Helix-mRNA"
 
     def __init__(

modeling_helix_mrna.py

@@ -1459,6 +1459,7 @@ class HelixmRNAModel(HelixmRNAPreTrainedModel):
             raise ValueError("Config must be provided")
 
         model_name = wrapper_config.model_name
+        kwargs.pop("trust_remote_code", None)  # don't pass to non-Auto config class (HelixmRNAConfig)
         cfg = HelixmRNAConfig.from_pretrained(model_name, **kwargs)
         cfg.model_name = model_name