CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-10-30 08:42:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Shangning Xu
2023-07-15 02:40:05 +08:00
committed by GitHub
parent 206e01de11
commit c48c525f87
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/embd-input/README.md
View File

@@ -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"]

2
examples/embd-input/llava.py
View File

@@ -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?")