Merge pull request kyegomez#100 from vyomakesh09/master

Former-commit-id: 4f68c54
ZackBradshaw · Nov 7, 2023 · f8630db · f8630db
2 parents 86362cd + 6b0d5d4
commit f8630db
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 = [