davanstrien HF Staff commited on
Commit
cd44e89
·
1 Parent(s): 6e6bac8

refactor: import vllm module for improved functionality

Browse files
Files changed (1) hide show
  1. generate_summaries_uv.py +1 -0
generate_summaries_uv.py CHANGED
@@ -27,6 +27,7 @@ from huggingface_hub import login, dataset_info, snapshot_download
27
  from tqdm.auto import tqdm
28
  from transformers import AutoTokenizer
29
  from vllm import LLM, SamplingParams
 
30
 
31
  # Setup logging
32
  logging.basicConfig(
 
27
  from tqdm.auto import tqdm
28
  from transformers import AutoTokenizer
29
  from vllm import LLM, SamplingParams
30
+ import vllm
31
 
32
  # Setup logging
33
  logging.basicConfig(