Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Remove TRANSFORMERS_CACHE
Browse files- Dockerfile +0 -1
- backend/app/core/cache.py +4 -4
Dockerfile
CHANGED
@@ -49,7 +49,6 @@ WORKDIR /app
|
|
49 |
|
50 |
# Environment variables
|
51 |
ENV HF_HOME=/app/.cache \
|
52 |
-
TRANSFORMERS_CACHE=/app/.cache \
|
53 |
HF_DATASETS_CACHE=/app/.cache \
|
54 |
INTERNAL_API_PORT=7861 \
|
55 |
PORT=7860 \
|
|
|
49 |
|
50 |
# Environment variables
|
51 |
ENV HF_HOME=/app/.cache \
|
|
|
52 |
HF_DATASETS_CACHE=/app/.cache \
|
53 |
INTERNAL_API_PORT=7861 \
|
54 |
PORT=7860 \
|
backend/app/core/cache.py
CHANGED
@@ -3,7 +3,7 @@ import shutil
|
|
3 |
from pathlib import Path
|
4 |
from datetime import timedelta
|
5 |
import logging
|
6 |
-
from app.
|
7 |
from app.config.base import (
|
8 |
CACHE_ROOT,
|
9 |
DATASETS_CACHE,
|
@@ -58,16 +58,16 @@ class CacheConfig:
|
|
58 |
def _setup_environment(self):
|
59 |
"""Configure HuggingFace environment variables"""
|
60 |
logger.info(LogFormatter.subsection("ENVIRONMENT SETUP"))
|
61 |
-
|
62 |
env_vars = {
|
63 |
"HF_HOME": str(self.cache_root),
|
64 |
-
"TRANSFORMERS_CACHE": str(self.models_cache),
|
65 |
"HF_DATASETS_CACHE": str(self.datasets_cache)
|
66 |
}
|
67 |
-
|
68 |
for var, value in env_vars.items():
|
69 |
os.environ[var] = value
|
70 |
logger.info(LogFormatter.info(f"Set {var}={value}"))
|
|
|
71 |
|
72 |
def get_cache_path(self, cache_type: str) -> Path:
|
73 |
"""Returns the path for a specific cache type"""
|
|
|
3 |
from pathlib import Path
|
4 |
from datetime import timedelta
|
5 |
import logging
|
6 |
+
from app.core.formatting import LogFormatter
|
7 |
from app.config.base import (
|
8 |
CACHE_ROOT,
|
9 |
DATASETS_CACHE,
|
|
|
58 |
def _setup_environment(self):
|
59 |
"""Configure HuggingFace environment variables"""
|
60 |
logger.info(LogFormatter.subsection("ENVIRONMENT SETUP"))
|
61 |
+
|
62 |
env_vars = {
|
63 |
"HF_HOME": str(self.cache_root),
|
|
|
64 |
"HF_DATASETS_CACHE": str(self.datasets_cache)
|
65 |
}
|
66 |
+
|
67 |
for var, value in env_vars.items():
|
68 |
os.environ[var] = value
|
69 |
logger.info(LogFormatter.info(f"Set {var}={value}"))
|
70 |
+
|
71 |
|
72 |
def get_cache_path(self, cache_type: str) -> Path:
|
73 |
"""Returns the path for a specific cache type"""
|