高橋慧 commited on
Commit
f1c8bd8
·
1 Parent(s): e096153

add docker and change llama model

Browse files
Files changed (2) hide show
  1. Dockerfile +21 -0
  2. app.py +1 -1
Dockerfile ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.10-bookworm # 安定のため Debian 12 に固定するのがおすすめ
2
+
3
+ RUN apt-get update && apt-get install -y \
4
+ git \
5
+ git-lfs \
6
+ ffmpeg \
7
+ libsm6 \
8
+ libxext6 \
9
+ cmake \
10
+ rsync \
11
+ libgl1 \
12
+ libglib2.0-0 \
13
+ && rm -rf /var/lib/apt/lists/* \
14
+ && git lfs install
15
+
16
+ WORKDIR /home/user/app
17
+ COPY requirements.txt ./
18
+ RUN pip install --no-cache-dir -r requirements.txt
19
+
20
+ COPY . .
21
+ CMD ["python", "app.py"]
app.py CHANGED
@@ -10,7 +10,7 @@ import gradio as gr
10
 
11
  #models
12
  LLAMA3_8B = "Llama3-8b-8192"
13
- LLAMA3_70B = "Llama3-70b-8192"
14
  Mixtral = "mixtral-8x7b-32768"
15
 
16
  def custom_completion_to_prompt(completion: str) -> str:
 
10
 
11
  #models
12
  LLAMA3_8B = "Llama3-8b-8192"
13
+ LLAMA3_70B = "llama-3.1-8b-instant"
14
  Mixtral = "mixtral-8x7b-32768"
15
 
16
  def custom_completion_to_prompt(completion: str) -> str: