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

messagers/message_composer.py

@@ -126,7 +126,7 @@ class MessageComposer:
             self.merged_str_list.append(f"GPT4 Correct Assistant:\n")
             self.merged_str = "\n".join(self.merged_str_list)
         # https://huggingface.co/google/gemma-1.1-7b-it#chat-template
-        elif self.model in ["gemma-7b"]:
+        elif self.model in ["gemma-1.1-7b"]:
             self.messages = self.concat_messages_by_role(messages)
             self.merged_str_list = []
             self.end_of_turn = "<end_of_turn>"
@@ -163,7 +163,7 @@ class MessageComposer:
 if __name__ == "__main__":
     # model = "mixtral-8x7b"
     # model = "nous-mixtral-8x7b"
-    # model = "gemma-7b"
+    # model = "gemma-1.1-7b"
     model = "openchat-3.5"
     composer = MessageComposer(model)
     messages = [