bugfix: invalid path on LatentReceiver

This commit is contained in:
Dr.Lt.Data
2023-06-27 01:29:46 +09:00
parent ec249bd5e3
commit c711aa7018
2 changed files with 16 additions and 6 deletions
+1 -1
View File
@@ -1,7 +1,7 @@
import configparser
import os
version = "V2.15.1"
version = "V2.15.2"
dependency_version = 1
+15 -5
View File
@@ -8,6 +8,7 @@ from segment_anything import sam_model_registry
from io import BytesIO
import piexif
import zipfile
import re
from server import PromptServer
from impact.utils import *
@@ -1539,6 +1540,19 @@ class LatentReceiver:
return {"samples": tensor['latent_tensor']}
return None
def parse_filename(self, filename):
pattern = r"^(.*)/(.*?)\[(.*)\]\s*$"
match = re.match(pattern, filename)
if match:
subfolder = match.group(1)
filename = match.group(2).rstrip() # 오른쪽 공백 제거
file_type = match.group(3)
else:
subfolder = ''
file_type = self.type
return {'filename': filename, 'subfolder': subfolder, 'type': file_type}
def doit(self, latent, link_id):
latent_path = folder_paths.get_annotated_filepath(latent)
@@ -1551,11 +1565,7 @@ class LatentReceiver:
else:
samples = LatentReceiver.load_preview_latent(latent_path)
preview = {
'filename': latent_path,
'subfolder': '',
'type': self.type
}
preview = self.parse_filename(latent)
return {
'ui': {"images": [preview]},