Spaces:
Paused
Paused
Husnain
commited on
♻️ [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"])
|