Mirrors/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-10-30 06:30:15 +01: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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats 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?")