add client, config to input_supported

This commit is contained in:
Believethehype
2023-12-28 22:56:41 +01:00
parent 4c07beff7b
commit bc3701d498
21 changed files with 28 additions and 23 deletions

View File

@@ -33,7 +33,7 @@ class ImageGenerationReplicateSDXL(DVMTaskInterface):
dvm_config.SCRIPT = os.path.abspath(__file__)
super().__init__(name, dvm_config, nip89config, admin_config, options)
def is_input_supported(self, tags):
def is_input_supported(self, tags, client=None, dvm_config=None):
for tag in tags:
if tag.as_vec()[0] == 'i':
input_value = tag.as_vec()[1]