MLCD-Seg / vision_resampler.py
killTheHostage's picture
Update a convenient way to use this model through Huggingface
b74e18c
raw
history blame contribute delete
374 Bytes
'''
CopyRight @DeepGlint 2025
'''
import torch
def build_vision_resampler(model_args, delay_load=False, **kwargs):
return IdentityMap()
class IdentityMap(torch.nn.Module):
def __init__(self):
super().__init__()
def forward(self, x, *args, **kwargs):
return x
@property
def config(self):
return {"mm_resampler_type": None}