File size: 136 Bytes
c5ca37a
 
 
 
 
 
1
2
3
4
5
6

export GPU_ID=0

CUDA_VISIBLE_DEVICES=$GPU_ID python examples/run_generation.py \
    --model_type=gpt2 \
    --model_name_or_path=gpt2