Modify Architecture to match PR content (left out field)

arcee-ai · Jul 10, 2024 · 3c56b7b · 3c56b7b
1 parent 8e3fd71
commit 3c56b7b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mergekit/architecture.py b/mergekit/architecture.py
@@ -53,6 +53,7 @@ class WeightInfo(BaseModel, frozen=True):
     output_space: Optional[str] = None
     head_space: Optional[str] = None
     optional: bool = False
+    force_dtype: Optional[str] = None
     aliases: Optional[List[str]] = None
     head_split: Literal[None, "input", "output"] = None
     is_kq: Optional[bool] = False