Merge pull request #100 from vyomakesh09/master

Former-commit-id: 4f68c54cad
1 year ago · f8630db4d2
parent 86362cdaa0 6b0d5d4a4f
commit f8630db4d2
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 = [