File size: 786 Bytes
9fd1204 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from typing import Any, Dict, List, Optional
from .base import ProcessorMixin
from .canny import CannyProcessor
from .clip import CLIPPooledProcessor
from .glm import CogView4GLMProcessor
from .llama import LlamaProcessor
from .t5 import T5Processor
from .text import CaptionEmbeddingDropoutProcessor, CaptionTextDropoutProcessor
class CopyProcessor(ProcessorMixin):
r"""Processor that copies the input data unconditionally to the output."""
def __init__(self, output_names: List[str] = None, input_names: Optional[Dict[str, Any]] = None):
super().__init__()
self.output_names = output_names
self.input_names = input_names
assert len(output_names) == 1
def forward(self, input: Any) -> Any:
return {self.output_names[0]: input}
|