diff --git a/kohya_gui/blip2_caption_gui.py b/kohya_gui/blip2_caption_gui.py index 6a859af..036211c 100644 --- a/kohya_gui/blip2_caption_gui.py +++ b/kohya_gui/blip2_caption_gui.py @@ -23,7 +23,7 @@ def load_model(): # Initialize the BLIP2 processor processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b") - print('processor:' + str(processor)) + log.debug('Processor initialized: %s', processor) # Initialize the BLIP2 model model = Blip2ForConditionalGeneration.from_pretrained(