works with latest flax version 0.5.2, updated requirements.txt

This commit is contained in:
Brett Kuprel
2022-06-29 11:46:19 -04:00
parent c4f613c89f
commit 764b5bbc0e
3 changed files with 5 additions and 5 deletions
+2 -2
View File
@@ -12,8 +12,8 @@ parser.set_defaults(mega=False)
parser.add_argument('--torch', action='store_true')
parser.add_argument('--no-torch', dest='torch', action='store_false')
parser.set_defaults(torch=False)
parser.add_argument('--text', type=str)
parser.add_argument('--seed', type=int, default=0)
parser.add_argument('--text', type=str, default='alien life')
parser.add_argument('--seed', type=int, default=7)
parser.add_argument('--image_path', type=str, default='generated')
parser.add_argument('--sample_token_count', type=int, default=256) # for debugging