add AnyText

This commit is contained in:
Qing
2024-01-21 23:25:50 +08:00
parent f5bd697687
commit 1905743886
17 changed files with 513 additions and 276 deletions

View File

@@ -254,7 +254,7 @@ class DDIMSampler(object):
)
img, pred_x0 = outs
if callback:
callback(i)
callback(None, i, None, None)
if img_callback:
img_callback(pred_x0, i)