From 0031a45a68403dabadf1cc386c9b0530f057aebf 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
Former-commit-id: 9c48ecdf906244caaf19b3bd9354727adf7ae315
---
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 = [