You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+10-5Lines changed: 10 additions & 5 deletions
Original file line number
Diff line number
Diff line change
@@ -110,7 +110,8 @@ decoder = Decoder(
110
110
unet= unet,
111
111
clip= clip,
112
112
timesteps=100,
113
-
cond_drop_prob=0.2
113
+
image_cond_drop_prob=0.1,
114
+
text_cond_drop_prob=0.5
114
115
).cuda()
115
116
116
117
# mock images (get a lot of this)
@@ -229,7 +230,8 @@ decoder = Decoder(
229
230
unet= (unet1, unet2), # insert both unets in order of low resolution to highest resolution (you can have as many stages as you want here)
230
231
image_sizes= (256, 512), # resolutions, 256 for first unet, 512 for second. these must be unique and in ascending order (matches with the unets passed in)
231
232
timesteps=1000,
232
-
cond_drop_prob=0.2
233
+
image_cond_drop_prob=0.1,
234
+
text_cond_drop_prob=0.5
233
235
).cuda()
234
236
235
237
# mock images (get a lot of this)
@@ -348,7 +350,8 @@ decoder = Decoder(
348
350
image_sizes= (128, 256),
349
351
clip= clip,
350
352
timesteps=100,
351
-
cond_drop_prob=0.2,
353
+
image_cond_drop_prob=0.1,
354
+
text_cond_drop_prob=0.5,
352
355
condition_on_text_encodings=False# set this to True if you wish to condition on text during training and sampling
353
356
).cuda()
354
357
@@ -558,7 +561,8 @@ decoder = Decoder(
558
561
image_sizes= (128, 256),
559
562
clip= clip,
560
563
timesteps=100,
561
-
cond_drop_prob=0.2,
564
+
image_cond_drop_prob=0.1,
565
+
text_cond_drop_prob=0.5,
562
566
condition_on_text_encodings=False# set this to True if you wish to condition on text during training and sampling
563
567
).cuda()
564
568
@@ -669,7 +673,8 @@ decoder = Decoder(
669
673
unet= (unet1, unet2, unet3), # insert unets in order of low resolution to highest resolution (you can have as many stages as you want here)
670
674
image_sizes= (256, 512, 1024), # resolutions, 256 for first unet, 512 for second, 1024 for third
0 commit comments