Update DATASET_SaveImage.py

2024-07-05 11:27:17 +05:30
parent 42ed27fa9a
commit e7c33fff47
1 changed files with 2 additions and 1 deletions
@@ -1,7 +1,8 @@
 import os
 import json
 import numpy as np
-from PIL import Image, PngInfo
+from PIL import Image
+from PIL.PngImagePlugin import PngInfo
 from comfy.cli_args import args

 class DataSet_SaveImage: