These should not raise NotImplemented

This commit is contained in:
doctorpangloss 2024-06-19 13:40:42 -07:00
parent 92f435aba3
commit facf68e7b9

View File

@ -123,7 +123,7 @@ class BaseModel(torch.nn.Module):
return self.adm_channels > 0 return self.adm_channels > 0
def encode_adm(self, **kwargs): def encode_adm(self, **kwargs):
raise NotImplementedError return None
def extra_conds(self, **kwargs): def extra_conds(self, **kwargs):
out = {} out = {}
@ -491,7 +491,7 @@ class SD_X4Upscaler(BaseModel):
class IP2P(BaseModel): class IP2P(BaseModel):
def process_ip2p_image_in(self, image): def process_ip2p_image_in(self, image):
raise NotImplementedError return None
def extra_conds(self, **kwargs): def extra_conds(self, **kwargs):
out = {} out = {}