Support the new phi3 models (#530)

EricLBuehler · Jul 2, 2024 · 5ded151 · 5ded151
1 parent c834f59
commit 5ded151
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mistralrs-core/src/layers.rs b/mistralrs-core/src/layers.rs
@@ -84,7 +84,7 @@ impl FromStr for ScaledRopeType {
     type Err = candle_core::Error;
     fn from_str(s: &str) -> std::result::Result<Self, Self::Err> {
         match s {
-            "su" => Ok(Self::Su),
+            "su" | "longrope" => Ok(Self::Su),
             "yarn" => Ok(Self::Yarn),
             _ => Err(candle_core::Error::Msg(
                 "Expected either `su` or `yarn` scaled RoPE type.".to_string(),