Fixes for sdxl ip adapter training. Bug fixes

2023-12-21 11:15:58 -07:00
parent 0f597f453e
commit 7703e3a15e
5 changed files with 63 additions and 11 deletions
--- a/toolkit/image_utils.py
+++ b/toolkit/image_utils.py
@@ -432,7 +432,7 @@ def show_img(img, name='AI Toolkit'):

    img = np.clip(img, 0, 255).astype(np.uint8)
    cv2.imshow(name, img[:, :, ::-1])
-    k = cv2.waitKey(10) & 0xFF
+    k = cv2.waitKey(1) & 0xFF
    if k == 27:  # Esc key to stop
        print('\nESC pressed, stopping')
        raise KeyboardInterrupt