mirror of https://github.com/bmaltais/kohya_ss
Update kohya_gui/blip2_caption_gui.py
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>pull/3252/head
parent
59b4bb6977
commit
451f051d52
|
|
@ -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(
|
||||
|
|
|
|||
Loading…
Reference in New Issue