fix: typo bfloat16

Former-commit-id: 9c48ecdf906244caaf19b3bd9354727adf7ae315
2 years ago · 0031a45a68
parent 7e3bcc0a69
commit 0031a45a68
1 changed files with 1 additions and 1 deletions
--- a/swarms/models/zephyr.py
+++ b/swarms/models/zephyr.py
@ -39,7 +39,7 @@ class Zephyr:
        self.pipe = pipeline(
            "text-generation",
            model="HuggingFaceH4/zephyr-7b-alpha",
-            torch_dtype=torch.bfloa16,
+            torch_dtype=torch.bfloat16,
            device_map="auto",
        )
        self.messages = [