diff --git a/launcher/src/main.rs b/launcher/src/main.rs index e4d5bb85..72d0057e 100644 --- a/launcher/src/main.rs +++ b/launcher/src/main.rs @@ -148,6 +148,7 @@ impl std::fmt::Display for Dtype { enum RopeScaling { Linear, Dynamic, + Yarn, } impl std::fmt::Display for RopeScaling { @@ -160,6 +161,9 @@ impl std::fmt::Display for RopeScaling { RopeScaling::Dynamic => { write!(f, "dynamic") } + RopeScaling::Yarn => { + write!(f, "yarn") + } } } }