mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-14 20:58:08 +00:00
9 lines
286 B
Python
9 lines
286 B
Python
import torch
|
|
import torch.utils.checkpoint
|
|
|
|
from transformers.models.blip.image_processing_blip import BlipImageProcessor
|
|
|
|
|
|
class ImgprocModelImageProcessor(BlipImageProcessor):
|
|
def new_image_processing_method(self, pixel_values: torch.FloatTensor):
|
|
return pixel_values / 2
|