:recycle: [Refactor] Rename gemma-7b to gemma-1.1-7b

#3
Files changed (1) hide show
  1. networks/huggingface_streamer.py +2 -2
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: