ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: fail fast if runtime pip install of transformers 5.x fails

Browse source
This commit is contained in:
Roland Tannous 2026-03-07 08:40:25 +00:00
commit 2c4c598832
3 changed files with 24 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

12
studio/backend/core/export/worker.py
View file

@ -47,18 +47,22 @@ def _activate_transformers_version(model_name: str, project_root: str) -> None:
logger.warning(".venv_t5 not found at %s — installing at runtime", venv_t5)
import subprocess as sp
os.makedirs(venv_t5, exist_ok=True)
sp.run(
r1 = sp.run(
[sys.executable, "-m", "pip", "install", "--target", venv_t5,
"--no-deps", "transformers==5.1.0"],
stdout=sp.PIPE, stderr=sp.STDOUT,
)
sp.run(
r2 = sp.run(
[sys.executable, "-m", "pip", "install", "--target", venv_t5,
"--no-deps", "huggingface_hub==1.3.0"],
stdout=sp.PIPE, stderr=sp.STDOUT,
)
if os.path.isdir(venv_t5):
sys.path.insert(0, venv_t5)
if r1.returncode != 0 or r2.returncode != 0:
raise RuntimeError(
f"Failed to install transformers 5.x into {venv_t5}. "
f"pip returncode: transformers={r1.returncode}, huggingface_hub={r2.returncode}"
)
sys.path.insert(0, venv_t5)
else:
logger.info("Using default transformers (4.57.x) for %s", model_name)

12
studio/backend/core/inference/worker.py
View file

@ -49,18 +49,22 @@ def _activate_transformers_version(model_name: str, project_root: str) -> None:
logger.warning(".venv_t5 not found at %s — installing at runtime", venv_t5)
import subprocess as sp
os.makedirs(venv_t5, exist_ok=True)
sp.run(
r1 = sp.run(
[sys.executable, "-m", "pip", "install", "--target", venv_t5,
"--no-deps", "transformers==5.1.0"],
stdout=sp.PIPE, stderr=sp.STDOUT,
)
sp.run(
r2 = sp.run(
[sys.executable, "-m", "pip", "install", "--target", venv_t5,
"--no-deps", "huggingface_hub==1.3.0"],
stdout=sp.PIPE, stderr=sp.STDOUT,
)
if os.path.isdir(venv_t5):
sys.path.insert(0, venv_t5)
if r1.returncode != 0 or r2.returncode != 0:
raise RuntimeError(
f"Failed to install transformers 5.x into {venv_t5}. "
f"pip returncode: transformers={r1.returncode}, huggingface_hub={r2.returncode}"
)
sys.path.insert(0, venv_t5)
else:
logger.info("Using default transformers (4.57.x) for %s", model_name)

12
studio/backend/core/training/worker.py
View file

@ -44,18 +44,22 @@ def _activate_transformers_version(model_name: str, project_root: str) -> None:
logger.warning(".venv_t5 not found at %s — installing at runtime", venv_t5)
import subprocess as sp
os.makedirs(venv_t5, exist_ok=True)
sp.run(
r1 = sp.run(
[sys.executable, "-m", "pip", "install", "--target", venv_t5,
"--no-deps", "transformers==5.1.0"],
stdout=sp.PIPE, stderr=sp.STDOUT,
)
sp.run(
r2 = sp.run(
[sys.executable, "-m", "pip", "install", "--target", venv_t5,
"--no-deps", "huggingface_hub==1.3.0"],
stdout=sp.PIPE, stderr=sp.STDOUT,
)
if os.path.isdir(venv_t5):
sys.path.insert(0, venv_t5)
if r1.returncode != 0 or r2.returncode != 0:
raise RuntimeError(
f"Failed to install transformers 5.x into {venv_t5}. "
f"pip returncode: transformers={r1.returncode}, huggingface_hub={r2.returncode}"
)
sys.path.insert(0, venv_t5)
else:
logger.info("Using default transformers (4.57.x) for %s", model_name)