ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

AutoModelForImageTextToText

Browse source
This commit is contained in:
Daniel Han 2025-03-11 04:30:01 -07:00
commit 2015e1e382
2 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
unsloth/models/loader.py
View file

@ -428,9 +428,15 @@ from ..kernels import (
)
from .vision import FastBaseModel
from transformers import (
AutoModelForVision2Seq,
AutoModelForCausalLM,
)
try:
from transformers import AutoModelForImageTextToText
AutoModelForVision2Seq = AutoModelForImageTextToText
except:
from transformers import AutoModelForVision2Seq
pass
class FastModel(FastBaseModel):
@staticmethod

7
unsloth/models/vision.py
View file

@ -15,11 +15,16 @@
import torch
from transformers import (
BitsAndBytesConfig,
AutoModelForVision2Seq,
AutoProcessor,
AutoTokenizer,
AutoModelForCausalLM,
)
try:
from transformers import AutoModelForImageTextToText
AutoModelForVision2Seq = AutoModelForImageTextToText
except:
from transformers import AutoModelForVision2Seq
pass
from .llama import *
from ..kernels import (
post_patch_loss_function,