fix: typo bfloat16

kyegomez · Nov 7, 2023 · b61fcf3 · b61fcf3
1 parent 16176e8
commit b61fcf3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swarms/models/zephyr.py b/swarms/models/zephyr.py
@@ -39,7 +39,7 @@ def __init__(
         self.pipe = pipeline(
             "text-generation",
             model="HuggingFaceH4/zephyr-7b-alpha",
-            torch_dtype=torch.bfloa16,
+            torch_dtype=torch.bfloat16,
             device_map="auto",
         )
         self.messages = [