davanstrien HF Staff commited on
Commit
58454eb
·
1 Parent(s): 9fa7761

don't set flashinfer

Browse files
Files changed (1) hide show
  1. 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 \\")