update name

HelloiOS2014 · May 23, 2024 · 43e7097 · 43e7097
1 parent 6dde8b9
commit 43e7097
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 10 deletions.
diff --git a/README.md b/README.md
@@ -492,13 +492,13 @@ Please refer to the following codes to run.
 
 
 ```python
-from chat import OmniLMMChat, img2base64
+from chat import MiniCPMVChat, img2base64
 import torch
 import json
 
 torch.manual_seed(0)
 
-chat_model = OmniLMMChat('openbmb/MiniCPM-Llama3-V-2_5')
+chat_model = MiniCPMVChat('openbmb/MiniCPM-Llama3-V-2_5')
 
 im_64 = img2base64('./assets/airplane.jpeg')
 

diff --git a/README_en.md b/README_en.md
@@ -470,13 +470,13 @@ Please refer to the following codes to run `MiniCPM-V` and `OmniLMM`.
 
 
 ```python
-from chat import OmniLMMChat, img2base64
+from chat import MiniCPMVChat, img2base64
 import torch
 import json
 
 torch.manual_seed(0)
 
-chat_model = OmniLMMChat('openbmb/MiniCPM-Llama3-V-2_5')
+chat_model = MiniCPMVChat('openbmb/MiniCPM-Llama3-V-2_5')
 
 im_64 = img2base64('./assets/airplane.jpeg')
 

diff --git a/README_zh.md b/README_zh.md
@@ -506,13 +506,13 @@ pip install -r requirements.txt
 
 
 ```python
-from chat import OmniLMMChat, img2base64
+from chat import MiniCPMVChat, img2base64
 import torch
 import json
 
 torch.manual_seed(0)
 
-chat_model = OmniLMMChat('openbmb/MiniCPM-Llama3-V-2_5')
+chat_model = MiniCPMVChat('openbmb/MiniCPM-Llama3-V-2_5')
 
 im_64 = img2base64('./assets/airplane.jpeg')
 

diff --git a/chat.py b/chat.py
@@ -136,7 +136,7 @@ def img2base64(file_name):
         encoded_string = base64.b64encode(f.read())
         return encoded_string
 
-class OmniLMM3B:
+class MiniCPMV:
     def __init__(self, model_path) -> None:
         self.model = AutoModel.from_pretrained(model_path, trust_remote_code=True).to(dtype=torch.bfloat16)
         self.tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
@@ -184,14 +184,14 @@ def chat(self, input):
         return answer
 
 
-class OmniLMMChat:
+class MiniCPMVChat:
     def __init__(self, model_path) -> None:
         if '12B' in model_path:
             self.model = OmniLMM12B(model_path)
         elif 'MiniCPM-Llama3-V' in model_path:
             self.model = MiniCPMV2_5(model_path)
         else:
-            self.model = OmniLMM3B(model_path)
+            self.model = MiniCPMV(model_path)
 
     def chat(self, input):
         return self.model.chat(input)
@@ -200,7 +200,7 @@ def chat(self, input):
 if __name__ == '__main__':
 
     model_path = 'openbmb/OmniLMM-12B'
-    chat_model = OmniLMMChat(model_path)
+    chat_model = MiniCPMVChat(model_path)
 
     im_64 = img2base64('./assets/worldmap_ck.jpg')