Spaces:
Running
Running
:recycle: [Refactor] Rename gemma-7b to gemma-1.1-7b
#3
by
Niansuh
- opened
networks/huggingface_streamer.py
CHANGED
@@ -25,8 +25,8 @@ class HuggingfaceStreamer:
|
|
25 |
self.model_fullname = MODEL_MAP[self.model]
|
26 |
self.message_outputer = OpenaiStreamOutputer(model=self.model)
|
27 |
|
28 |
-
if self.model == "gemma-7b":
|
29 |
-
# this is not wrong, as repo `google/gemma-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"])
|
32 |
else:
|
|
|
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"])
|
32 |
else:
|