File generate_vllm.py

data/generation/generate_vllm.py:None–None · view source on GitHub ↗

Source from the content-addressed store, hash-verified

1	import os
2	import argparse
3	from data_utils import get_gen_dataset
4	from transformers import GenerationConfig, AutoModelForCausalLM, AutoTokenizer

nothing calls this directly

mainFunction · 0.70

no test coverage detected