Merge pull request #100 from vyomakesh09/master

Former-commit-id: eacc90be17bf65954e9333823ecca1b3480007e6
2 years ago · ddfca49300
parent 7e3bcc0a69 0031a45a68
commit ddfca49300
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 = [