From 3c56b7bf07606a85b3fdc98d61f50bbdebfe279f Mon Sep 17 00:00:00 2001
From: Luke Meyers <functor.soup@gmail.com>
Date: Wed, 10 Jul 2024 14:21:19 -0400
Subject: [PATCH] Modify Architecture to match PR content (left out field)

---
 mergekit/architecture.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mergekit/architecture.py b/mergekit/architecture.py
index 3bccaf16..b063362e 100644
--- 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