Skip to content

Inpaint

dataphy.visionpack.adapters.inpaint

Classes

InpaintAdapter(prompt: str = '', generate_once: bool = True)

Bases: Module

Source code in src/dataphy/visionpack/adapters/inpaint.py 
5
6
7
def __init__(self, prompt: str = "", generate_once: bool = True):
    super().__init__()
    self.prompt, self.generate_once = prompt, generate_once
Functions
forward(batch)
Source code in src/dataphy/visionpack/adapters/inpaint.py 
 9
10
11
def forward(self, batch):
    # TODO: Grounding + SAM + Diffusers pipeline
    return batch