зеркало из https://github.com/Azure/AML-AirField.git
42 строки
1.4 KiB
Bash
42 строки
1.4 KiB
Bash
|
set -e
|
||
|
|
|
||
|
|
if [ ! -d ./torchmoji ]; then
|
||
|
|
echo ""
|
||
|
|
echo Downloading Model
|
||
|
|
echo ""
|
||
|
|
|
||
|
|
# Clone torchMoji repo
|
||
|
|
git clone https://github.com/huggingface/torchMoji.git ./torchmoji_repo
|
||
|
|
|
||
|
|
# Copy model related files over to the model directory
|
||
|
|
mkdir -p torchmoji
|
||
|
|
cp ./torchmoji_repo/torchmoji/attlayer.py ./torchmoji/
|
||
|
|
cp ./torchmoji_repo/torchmoji/create_vocab.py ./torchmoji/
|
||
|
|
cp ./torchmoji_repo/torchmoji/filter_utils.py ./torchmoji/
|
||
|
|
cp ./torchmoji_repo/torchmoji/global_variables.py ./torchmoji/
|
||
|
|
cp ./torchmoji_repo/torchmoji/lstm.py ./torchmoji/
|
||
|
|
cp ./torchmoji_repo/torchmoji/model_def.py ./torchmoji/
|
||
|
|
cp ./torchmoji_repo/torchmoji/sentence_tokenizer.py ./torchmoji/
|
||
|
|
cp ./torchmoji_repo/torchmoji/tokenizer.py ./torchmoji/
|
||
|
|
cp ./torchmoji_repo/torchmoji/word_generator.py ./torchmoji/
|
||
|
|
cp ./torchmoji_repo/LICENSE ./torchmoji/
|
||
|
|
|
||
|
|
mkdir -p data
|
||
|
|
cp ./torchmoji_repo/data/emoji_codes.json ./data/
|
||
|
|
cp ./torchmoji_repo/data/filtering/wanted_emojis.csv ./data/
|
||
|
|
|
||
|
|
mkdir -p model
|
||
|
|
cp ./torchmoji_repo/model/vocabulary.json ./model/
|
||
|
|
|
||
|
|
echo ""
|
||
|
|
cat legal.notice
|
||
|
|
echo ""
|
||
|
|
read -n1 -s -r -p "Press any key to continue . . ."
|
||
|
|
echo ""
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [ ! -f model/pytorch_model.bin ]; then
|
||
|
|
# Download model weights
|
||
|
|
wget -nc https://www.dropbox.com/s/q8lax9ary32c7t9/pytorch_model.bin?dl=0# -O model/pytorch_model.bin
|
||
|
|
fi
|