瀏覽代碼

examples : fixed path typos in embd-input (#2214)

Shangning Xu 2 年之前
父節點
當前提交
c48c525f87
共有 2 個文件被更改,包括 2 次插入2 次删除
  1. 1 1
      examples/embd-input/README.md
  2. 1 1
      examples/embd-input/llava.py

+ 1 - 1
examples/embd-input/README.md

@@ -17,7 +17,7 @@ make
 import torch
 import torch
 
 
 bin_path = "../LLaVA-13b-delta-v1-1/pytorch_model-00003-of-00003.bin"
 bin_path = "../LLaVA-13b-delta-v1-1/pytorch_model-00003-of-00003.bin"
-pth_path = "./examples/embd_input/llava_projection.pth"
+pth_path = "./examples/embd-input/llava_projection.pth"
 
 
 dic = torch.load(bin_path)
 dic = torch.load(bin_path)
 used_key = ["model.mm_projector.weight","model.mm_projector.bias"]
 used_key = ["model.mm_projector.weight","model.mm_projector.bias"]

+ 1 - 1
examples/embd-input/llava.py

@@ -59,7 +59,7 @@ if __name__=="__main__":
     # Also here can use pytorch_model-00003-of-00003.bin directly.
     # Also here can use pytorch_model-00003-of-00003.bin directly.
     a.load_projection(os.path.join(
     a.load_projection(os.path.join(
         os.path.dirname(__file__) ,
         os.path.dirname(__file__) ,
-        "llava_projetion.pth"))
+        "llava_projection.pth"))
     respose = a.chat_with_image(
     respose = a.chat_with_image(
         Image.open("./media/llama1-logo.png").convert('RGB'),
         Image.open("./media/llama1-logo.png").convert('RGB'),
         "what is the text in the picture?")
         "what is the text in the picture?")