From 036a5a02a7a3e022399bdeeaa01f65b84d8f316b Mon Sep 17 00:00:00 2001
From: calycekr <4094424+calycekr@users.noreply.github.com>
Date: Mon, 17 Jun 2024 21:52:52 +0900
Subject: [PATCH] Fix missing rope scaling option for YaRN

---
 launcher/src/main.rs | 4 ++++
 1 file changed, 4 insertions(+)

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")
+            }
         }
     }
 }