Unverified Commit f9ee46f7 authored by Billel Mokeddem's avatar Billel Mokeddem Committed by GitHub
Browse files

Fix: Change model_type from ssm to mamba (#2740)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-28-135.us-west-2.compute.internal>
parent 8442f1ac
...@@ -534,7 +534,7 @@ def get_model( ...@@ -534,7 +534,7 @@ def get_model(
# TODO: fix how we determine model type for Mamba # TODO: fix how we determine model type for Mamba
if "ssm_cfg" in config_dict: if "ssm_cfg" in config_dict:
# *only happens in Mamba case # *only happens in Mamba case
model_type = "ssm" model_type = "mamba"
else: else:
raise RuntimeError( raise RuntimeError(
f"Could not determine model type for {model_id} revision {revision}" f"Could not determine model type for {model_id} revision {revision}"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment