Spaces:
Running
Running
limcheekin
commited on
Commit
•
237b081
1
Parent(s):
e29a647
feat: added clip model and chat_format
Browse files- Dockerfile +2 -1
- main.py +3 -1
Dockerfile
CHANGED
@@ -15,7 +15,8 @@ RUN pip install -U pip setuptools wheel && \
|
|
15 |
|
16 |
# Download model
|
17 |
RUN mkdir model && \
|
18 |
-
curl -L https://huggingface.co/abetlen/BakLLaVA-1-GGUF/resolve/main/bakllava-1.Q6_K.gguf -o model/gguf-model.bin
|
|
|
19 |
|
20 |
COPY ./start_server.sh ./
|
21 |
COPY ./main.py ./
|
|
|
15 |
|
16 |
# Download model
|
17 |
RUN mkdir model && \
|
18 |
+
curl -L https://huggingface.co/abetlen/BakLLaVA-1-GGUF/resolve/main/bakllava-1.Q6_K.gguf -o model/gguf-model.bin && \
|
19 |
+
curl -L https://huggingface.co/abetlen/BakLLaVA-1-GGUF/resolve/main/mmproj-model-f16.gguf -o model/clip-model.bin
|
20 |
|
21 |
COPY ./start_server.sh ./
|
22 |
COPY ./main.py ./
|
main.py
CHANGED
@@ -6,7 +6,9 @@ app = create_app(
|
|
6 |
Settings(
|
7 |
n_threads=2, # set to number of cpu cores
|
8 |
model="model/gguf-model.bin",
|
9 |
-
|
|
|
|
|
10 |
)
|
11 |
)
|
12 |
|
|
|
6 |
Settings(
|
7 |
n_threads=2, # set to number of cpu cores
|
8 |
model="model/gguf-model.bin",
|
9 |
+
clip_model_path="model/clip-model.bin",
|
10 |
+
embedding=True,
|
11 |
+
chat_format="llava-1-5"
|
12 |
)
|
13 |
)
|
14 |
|