sam_extension/pipeline/base.py

import torch
from torch import nn
from typing import Union, Dict
from dataclasses import dataclass

@dataclass(repr=True)
class Output:
    pass

class Pipeline(nn.Module):
    def __init__(self, *args, **kwargs):
        super(Pipeline, self).__init__()
        self.args = args
        self.kwargs = kwargs
    @classmethod
    def from_pretrained(cls, ckpt_path, device='cuda', *args, **kwargs):
        pass
    def forward(self, *args, **kwargs):
        pass