Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #33 from nawnie/patch-2 

train.py
This commit is contained in:
Victor Hall 2023-01-25 10:36:50 -08:00 committed by GitHub
parent b4847d6f45 e489e94d80
commit 9a9909c7ea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Train_Colab.ipynb
View File

@ -354,7 +354,7 @@
"if Disable_text_Encoder:\n", "if Disable_text_Encoder:\n",
" textencode = \"--disable_textenc_training Train_text \"\n", " textencode = \"--disable_textenc_training Train_text \"\n",
"\n", "\n",
"!python train_colab.py --resume_ckpt \"$model\" \\\n", "!python train.py --resume_ckpt \"$model\" \\\n",
" $textencode \\\n", " $textencode \\\n",
" $Gradient \\\n", " $Gradient \\\n",
" $shuffle \\\n", " $shuffle \\\n",
@ -378,8 +378,8 @@
" --save_every_n_epoch $Save_every_N_epoch \\\n", " --save_every_n_epoch $Save_every_N_epoch \\\n",
" --seed 555 \\\n", " --seed 555 \\\n",
" --shuffle_tags \\\n", " --shuffle_tags \\\n",
" --useadam8bit \\\n", " --useadam8bit \n",
" --notebook\n"
] ]
}, },
{ {