UL2-nemo-conversion / convert_hf_to_nemo.sh
Faton Rekathati
UL2 conversion instructions
0fd282e
# singularity exec --nv nemo2302 bash convert_hf_to_nemo.sh
OUTPUT_FOLDER=ul2-base-nl36-finnish
mkdir -p $OUTPUT_FOLDER
python hf_t5_v1_1_to_nemo.py \
--hf_model_name Finnish-NLP/ul2-base-nl36-finnish \
--nemo_state_dict $OUTPUT_FOLDER/nemo_state_dict.pt \
--nemo_file_path $OUTPUT_FOLDER/nemo_file.nemo