diff --git a/internal_controlnet/args.py b/internal_controlnet/args.py index 9090ded..7e6dcec 100644 --- a/internal_controlnet/args.py +++ b/internal_controlnet/args.py @@ -207,10 +207,10 @@ class ControlNetUnit(BaseModel): # the base64string will be interpret by torch.load to reconstruct ipadapter # preprocessor output. # Currently the option is only accessible in API calls. - ipadapter_input: Optional[List[torch.Tensor]] = None + ipadapter_input: Optional[List[Any]] = None @validator("ipadapter_input", pre=True) - def parse_ipadapter_input(cls, value) -> Optional[List[torch.Tensor]]: + def parse_ipadapter_input(cls, value) -> Optional[List[Any]]: if value is None: return None if isinstance(value, str):