Merge pull request #100 from vyomakesh09/master

kyegomez · Nov 7, 2023 · 4f68c54 · 4f68c54
2 parents 16176e8 + b61fcf3
commit 4f68c54
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 = [