Babel / Optimus /code /scripts /scripts_local /run_gpt_generation.sh
rynmurdock's picture
init
c5ca37a
export GPU_ID=0
CUDA_VISIBLE_DEVICES=$GPU_ID python examples/run_generation.py \
--model_type=gpt2 \
--model_name_or_path=gpt2