wassname/Clover-Edition
1
0
Fork 0
mirror of https://github.com/wassname/Clover-Edition.git synced 2026-09-09 11:13:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Nick Walton
2019-11-18 22:30:53 -07:00
parent 8be5dbb123
commit 207b3b7b11
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/gpt2/gpt2_generator.py
+1 -1
View File
@@ -17,7 +17,7 @@ class GPT2Generator:
self.top_k = top_k
self.top_p = top_p
self.model_name = "model_v3"
self.model_name = "model_v4"
self.model_dir = "generator/gpt2/models"
self.checkpoint_path = os.path.join(self.model_dir, self.model_name)
install.sh
+2 -2
View File
@@ -1,10 +1,10 @@
MODEL_DIRECTORY=aidungeon/generator/gpt2/models/model_v3
MODEL_DIRECTORY=aidungeon/generator/gpt2/models/model_v4
if [ -d "$MODEL_DIRECTORY" ]; then
echo "AIDungeon2 is already installed"
else
echo "Downloading AIDungeon2 Model"
gsutil -m cp -r gs://aidungeon2model/model_v3 ./generator/gpt2/models
gsutil -m cp -r gs://aidungeon2model/model_v4 ./generator/gpt2/models
pip install -r requirements.txt > /dev/null
fi