deal with new image nip96 links

This commit is contained in:
Believethehype 2023-12-29 15:13:39 +01:00
parent 17baf08d48
commit f4c1e834c6
3 changed files with 4 additions and 3 deletions

View File

@ -50,7 +50,7 @@ class MediaConverter(DVMTaskInterface):
if tag.as_vec()[0] == 'i':
input_type = tag.as_vec()[2]
if input_type == "url":
url = tag.as_vec()[1]
url = str(tag.as_vec()[1]).split('#')[0]
elif tag.as_vec()[0] == 'param':

View File

@ -89,7 +89,8 @@ class ImageGenerationSDXLIMG2IMG(DVMTaskInterface):
if input_type == "text":
prompt = tag.as_vec()[1]
elif input_type == "url":
url = tag.as_vec()[1]
url = str(tag.as_vec()[1]).split('#')[0]
elif tag.as_vec()[0] == 'param':
print("Param: " + tag.as_vec()[1] + ": " + tag.as_vec()[2])

View File

@ -60,7 +60,7 @@ class VideoGenerationReplicateSVD(DVMTaskInterface):
if tag.as_vec()[0] == 'i':
input_type = tag.as_vec()[2]
if input_type == "url":
url = tag.as_vec()[1]
url = str(tag.as_vec()[1]).split('#')[0]
# TODO add params as defined above
options = {