Merge pull request #100 from vyomakesh09/master

1 year ago · 4f68c54cad
parent 16176e8cad b61fcf3983
commit 4f68c54cad
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 = [