gpt2-medium-persian / src /run_config.sh
m3hrdadfi's picture
Add runner, fix some bugs
31bf2aa
raw
history blame
306 Bytes
#!/bin/bash
export LC_ALL=C.UTF-8
export LANG=C.UTF-8
export OUTPUT_DIR=./
#export OUTPUT_DIR=/home/username/code/gpt2-medium-persian
export NAME_OR_PATH=gpt2-medium
python src/create_config.py \
--output_dir="$OUTPUT_DIR" \
--name_or_path="$NAME_OR_PATH" \
--params='{"vocab_size": 50000}'