Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
58454eb
1
Parent(s):
9fa7761
don't set flashinfer
Browse files- generate_summaries_uv.py +2 -1
generate_summaries_uv.py
CHANGED
@@ -20,7 +20,7 @@ from typing import Optional
|
|
20 |
|
21 |
# Set environment variables to speed up model loading
|
22 |
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
|
23 |
-
os.environ["VLLM_ATTENTION_BACKEND"] = "FLASHINFER"
|
24 |
|
25 |
import polars as pl
|
26 |
from datasets import Dataset, load_dataset
|
@@ -231,6 +231,7 @@ if __name__ == "__main__":
|
|
231 |
print("export HOME=/tmp && \\")
|
232 |
print("export USER=dummy && \\")
|
233 |
print("export TORCHINDUCTOR_CACHE_DIR=/tmp/torch-inductor && \\")
|
|
|
234 |
print("uv run generate_summaries_uv.py \\")
|
235 |
print(" davanstrien/Smol-Hub-tldr \\")
|
236 |
print(" librarian-bots/dataset_cards_with_metadata \\")
|
|
|
20 |
|
21 |
# Set environment variables to speed up model loading
|
22 |
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
|
23 |
+
# os.environ["VLLM_ATTENTION_BACKEND"] = "FLASHINFER"
|
24 |
|
25 |
import polars as pl
|
26 |
from datasets import Dataset, load_dataset
|
|
|
231 |
print("export HOME=/tmp && \\")
|
232 |
print("export USER=dummy && \\")
|
233 |
print("export TORCHINDUCTOR_CACHE_DIR=/tmp/torch-inductor && \\")
|
234 |
+
print("export UV_TORCH_BACKEND=auto && \\")
|
235 |
print("uv run generate_summaries_uv.py \\")
|
236 |
print(" davanstrien/Smol-Hub-tldr \\")
|
237 |
print(" librarian-bots/dataset_cards_with_metadata \\")
|