Spaces:
Running
Running
Auto commit at 09-2025-08 0:00:01
Browse files- Dockerfile +8 -2
- app_huggingface.py +6 -2
- lily_llm_api/models/kanana_1_5_v_3b_instruct.py +5 -5
Dockerfile
CHANGED
|
@@ -37,15 +37,21 @@ RUN python -c "import nltk; nltk.download('punkt'); nltk.download('punkt_tab');
|
|
| 37 |
COPY . .
|
| 38 |
|
| 39 |
# ํ์ํ ๋๋ ํ ๋ฆฌ ์์ฑ
|
| 40 |
-
RUN mkdir -p /app/data /app/logs /app/models /app/uploads /app/vector_stores /app/temp /app/cache/transformers /app/cache/huggingface
|
| 41 |
|
| 42 |
-
# ๊ถํ ์ค์ - ์บ์ ๋๋ ํ ๋ฆฌ์ ์ฐ๊ธฐ ๊ถํ ๋ถ์ฌ
|
| 43 |
RUN chmod +x /app/*.py
|
| 44 |
RUN chmod -R 777 /app/cache
|
| 45 |
RUN chmod -R 777 /app/data
|
| 46 |
RUN chmod -R 777 /app/logs
|
| 47 |
RUN chmod -R 777 /app/uploads
|
| 48 |
RUN chmod -R 777 /app/temp
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
|
| 50 |
# Hugging Face ์บ์ ๋๋ ํ ๋ฆฌ ํ๊ฒฝ ๋ณ์ ์ค์
|
| 51 |
ENV TRANSFORMERS_CACHE=/app/cache/transformers
|
|
|
|
| 37 |
COPY . .
|
| 38 |
|
| 39 |
# ํ์ํ ๋๋ ํ ๋ฆฌ ์์ฑ
|
| 40 |
+
RUN mkdir -p /app/data /app/logs /app/models /app/uploads /app/vector_stores /app/temp /app/cache/transformers /app/cache/huggingface /app/cache/easyocr /app/cache/ocr_models
|
| 41 |
|
| 42 |
+
# ๊ถํ ์ค์ - ๋ชจ๋ ์บ์ ๋ฐ ๋ฐ์ดํฐ ๋๋ ํ ๋ฆฌ์ ์ฐ๊ธฐ ๊ถํ ๋ถ์ฌ
|
| 43 |
RUN chmod +x /app/*.py
|
| 44 |
RUN chmod -R 777 /app/cache
|
| 45 |
RUN chmod -R 777 /app/data
|
| 46 |
RUN chmod -R 777 /app/logs
|
| 47 |
RUN chmod -R 777 /app/uploads
|
| 48 |
RUN chmod -R 777 /app/temp
|
| 49 |
+
RUN chmod -R 777 /app/cache/easyocr
|
| 50 |
+
RUN chmod -R 777 /app/cache/ocr_models
|
| 51 |
+
|
| 52 |
+
# EasyOCR ๋ฐ ๊ธฐํ ๋ผ์ด๋ธ๋ฌ๋ฆฌ๋ฅผ ์ํ ํ๊ฒฝ ๋ณ์ ์ค์
|
| 53 |
+
ENV EASYOCR_MODULE_PATH=/app/cache/easyocr
|
| 54 |
+
ENV OCR_MODELS_PATH=/app/cache/ocr_models
|
| 55 |
|
| 56 |
# Hugging Face ์บ์ ๋๋ ํ ๋ฆฌ ํ๊ฒฝ ๋ณ์ ์ค์
|
| 57 |
ENV TRANSFORMERS_CACHE=/app/cache/transformers
|
app_huggingface.py
CHANGED
|
@@ -32,12 +32,14 @@ def setup_huggingface_environment():
|
|
| 32 |
# ํ์ํ ๋๋ ํ ๋ฆฌ ์์ฑ
|
| 33 |
directories = [
|
| 34 |
"data", "logs", "models", "uploads",
|
| 35 |
-
"vector_stores", "temp", "cache"
|
|
|
|
|
|
|
| 36 |
]
|
| 37 |
|
| 38 |
for dir_name in directories:
|
| 39 |
dir_path = project_root / dir_name
|
| 40 |
-
dir_path.mkdir(exist_ok=True)
|
| 41 |
logger.info(f"๐ ๋๋ ํ ๋ฆฌ ์์ฑ: {dir_path}")
|
| 42 |
|
| 43 |
# ํ๊ฒฝ ๋ณ์ ์ค์
|
|
@@ -45,6 +47,8 @@ def setup_huggingface_environment():
|
|
| 45 |
"TRANSFORMERS_CACHE": str(project_root / "cache" / "transformers"),
|
| 46 |
"HF_HOME": str(project_root / "cache" / "huggingface"),
|
| 47 |
"TORCH_HOME": str(project_root / "cache" / "torch"),
|
|
|
|
|
|
|
| 48 |
"TOKENIZERS_PARALLELISM": "false",
|
| 49 |
"OMP_NUM_THREADS": "1",
|
| 50 |
"MKL_NUM_THREADS": "1"
|
|
|
|
| 32 |
# ํ์ํ ๋๋ ํ ๋ฆฌ ์์ฑ
|
| 33 |
directories = [
|
| 34 |
"data", "logs", "models", "uploads",
|
| 35 |
+
"vector_stores", "temp", "cache",
|
| 36 |
+
"cache/transformers", "cache/huggingface",
|
| 37 |
+
"cache/easyocr", "cache/ocr_models"
|
| 38 |
]
|
| 39 |
|
| 40 |
for dir_name in directories:
|
| 41 |
dir_path = project_root / dir_name
|
| 42 |
+
dir_path.mkdir(exist_ok=True, parents=True)
|
| 43 |
logger.info(f"๐ ๋๋ ํ ๋ฆฌ ์์ฑ: {dir_path}")
|
| 44 |
|
| 45 |
# ํ๊ฒฝ ๋ณ์ ์ค์
|
|
|
|
| 47 |
"TRANSFORMERS_CACHE": str(project_root / "cache" / "transformers"),
|
| 48 |
"HF_HOME": str(project_root / "cache" / "huggingface"),
|
| 49 |
"TORCH_HOME": str(project_root / "cache" / "torch"),
|
| 50 |
+
"EASYOCR_MODULE_PATH": str(project_root / "cache" / "easyocr"),
|
| 51 |
+
"OCR_MODELS_PATH": str(project_root / "cache" / "ocr_models"),
|
| 52 |
"TOKENIZERS_PARALLELISM": "false",
|
| 53 |
"OMP_NUM_THREADS": "1",
|
| 54 |
"MKL_NUM_THREADS": "1"
|
lily_llm_api/models/kanana_1_5_v_3b_instruct.py
CHANGED
|
@@ -24,8 +24,8 @@ class Kanana15V3bInstructProfile:
|
|
| 24 |
self.local_path = "./lily_llm_core/models/kanana_1_5_v_3b_instruct"
|
| 25 |
self.display_name = "Kanana-1.5-v-3b-instruct (๋ก์ปฌ)"
|
| 26 |
else:
|
| 27 |
-
self.model_name = "gbrabbit/lily-math-model" # Hugging Face ๋ชจ๋ธ ๊ฒฝ๋ก
|
| 28 |
-
self.local_path =
|
| 29 |
self.display_name = "Kanana-1.5-v-3b-instruct (์๋ฒ)"
|
| 30 |
|
| 31 |
self.description = "์นด์นด์ค ๋ฉํฐ๋ชจ๋ฌ ๋ชจ๋ธ (3.6B) - Math RAG ํนํ"
|
|
@@ -146,9 +146,9 @@ class Kanana15V3bInstructProfile:
|
|
| 146 |
low_cpu_mem_usage=True,
|
| 147 |
).to(DEVICE)
|
| 148 |
else:
|
| 149 |
-
# Hugging Face Hub:
|
| 150 |
-
from transformers import
|
| 151 |
-
model =
|
| 152 |
model_path,
|
| 153 |
trust_remote_code=True,
|
| 154 |
torch_dtype=torch.float16,
|
|
|
|
| 24 |
self.local_path = "./lily_llm_core/models/kanana_1_5_v_3b_instruct"
|
| 25 |
self.display_name = "Kanana-1.5-v-3b-instruct (๋ก์ปฌ)"
|
| 26 |
else:
|
| 27 |
+
self.model_name = "gbrabbit/lily-math-model" # Hugging Face Hub ๋ชจ๋ธ ๊ฒฝ๋ก
|
| 28 |
+
self.local_path = None # ์๋ฒ์์๋ ๋ก์ปฌ ๊ฒฝ๋ก ์ฌ์ฉ ์ํจ
|
| 29 |
self.display_name = "Kanana-1.5-v-3b-instruct (์๋ฒ)"
|
| 30 |
|
| 31 |
self.description = "์นด์นด์ค ๋ฉํฐ๋ชจ๋ฌ ๋ชจ๋ธ (3.6B) - Math RAG ํนํ"
|
|
|
|
| 146 |
low_cpu_mem_usage=True,
|
| 147 |
).to(DEVICE)
|
| 148 |
else:
|
| 149 |
+
# Hugging Face Hub: AutoModel ์ฌ์ฉ (์ปค์คํ
์ค์ ํด๋์ค ํธํ์ฑ)
|
| 150 |
+
from transformers import AutoModel
|
| 151 |
+
model = AutoModel.from_pretrained(
|
| 152 |
model_path,
|
| 153 |
trust_remote_code=True,
|
| 154 |
torch_dtype=torch.float16,
|