Merge pull request kyegomez#100 from vyomakesh09/master

Former-commit-id: eacc90be17bf65954e9333823ecca1b3480007e6
ZackBradshaw · Nov 7, 2023 · ddfca49 · ddfca49
2 parents 7e3bcc0 + 0031a45
commit ddfca49
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 = [