Husnain
commited on
Commit
•
b1ed819
1
Parent(s):
4136636
♻️ [Refactor] Rename gemma-7b-1.1 to gemma-1.1-7b
Browse files
networks/huggingface_streamer.py
CHANGED
@@ -25,7 +25,7 @@ class HuggingfaceStreamer:
|
|
25 |
self.model_fullname = MODEL_MAP[self.model]
|
26 |
self.message_outputer = OpenaiStreamOutputer(model=self.model)
|
27 |
|
28 |
-
if self.model == "gemma-
|
29 |
# this is not wrong, as repo `google/gemma-1.1-7b-it` is gated and must authenticate to access it
|
30 |
# so I use mistral-7b as a fallback
|
31 |
self.tokenizer = AutoTokenizer.from_pretrained(MODEL_MAP["mistral-7b"])
|
|
|
25 |
self.model_fullname = MODEL_MAP[self.model]
|
26 |
self.message_outputer = OpenaiStreamOutputer(model=self.model)
|
27 |
|
28 |
+
if self.model == "gemma-1.1-7b":
|
29 |
# this is not wrong, as repo `google/gemma-1.1-7b-it` is gated and must authenticate to access it
|
30 |
# so I use mistral-7b as a fallback
|
31 |
self.tokenizer = AutoTokenizer.from_pretrained(MODEL_MAP["mistral-7b"])
|