Update app.py
Browse files
app.py
CHANGED
|
@@ -10,10 +10,6 @@ from transformers import Blip2Processor, Blip2ForConditionalGeneration, Instruct
|
|
| 10 |
|
| 11 |
def load_caption_model(blip2=False, instructblip=True):
|
| 12 |
|
| 13 |
-
|
| 14 |
-
model = YourModel()
|
| 15 |
-
|
| 16 |
-
|
| 17 |
if blip2:
|
| 18 |
processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16)
|
| 19 |
model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16)
|
|
|
|
| 10 |
|
| 11 |
def load_caption_model(blip2=False, instructblip=True):
|
| 12 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
if blip2:
|
| 14 |
processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16)
|
| 15 |
model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16)
|