diff --git a/studio/backend/requirements/extras-no-deps.txt b/studio/backend/requirements/extras-no-deps.txt index 4b5aa86b5f..9934bacd24 100644 --- a/studio/backend/requirements/extras-no-deps.txt +++ b/studio/backend/requirements/extras-no-deps.txt @@ -12,3 +12,5 @@ git+https://github.com/meta-pytorch/OpenEnv.git torch-c-dlpack-ext sentence_transformers==5.2.0 transformers==4.57.6 +pytorch_tokenizers +kernels diff --git a/studio/backend/requirements/overrides.txt b/studio/backend/requirements/overrides.txt index 6852f601ed..176c651b96 100644 --- a/studio/backend/requirements/overrides.txt +++ b/studio/backend/requirements/overrides.txt @@ -1,6 +1,2 @@ # Torch AO overrides (installed with --force-reinstall --no-cache-dir) torchao==0.14.0 -pytorch_tokenizers - -# Kernel packages -kernels