MLCD-Seg / vision_resampler.py

Update a convenient way to use this model through Huggingface

b74e18c over 1 year ago

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}

	'''
	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}