Husnain commited on
Commit
b1ed819
1 Parent(s): 4136636

♻️ [Refactor] Rename gemma-7b-1.1 to gemma-1.1-7b

Browse files
Files changed (1) hide show
  1. networks/huggingface_streamer.py +1 -1
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-7b-1.1":
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"])