From b61fcf3983dd0f8efa1e245ea4c7b7db461f5926 Mon Sep 17 00:00:00 2001
From: vyomakesh09 <vyomkeshkesh@gmail.com>
Date: Tue, 7 Nov 2023 10:48:08 +0000
Subject: [PATCH] fix: typo bfloat16

---
 swarms/models/zephyr.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/swarms/models/zephyr.py b/swarms/models/zephyr.py
index 582bc740..f4052d82 100644
--- 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 = [