This commit is contained in:
Nakarin Satayasai
2023-11-03 11:16:22 +07:00
parent 6dd3ff4c37
commit aab69eb5d5
+5 -3
View File
@@ -28,8 +28,10 @@ class CSVPromptsLoader():
},
}
RETURN_TYPES = ("STRING", "STRING", "STRING", "STRING", "STRING", )
RETURN_NAMES = ("styles", "face", "cloth", "action", "face_action",)
RETURN_TYPES = ("STRING", "STRING", "STRING",
"STRING", "STRING", "STRING",)
RETURN_NAMES = ("styles", "face", "cloth", "action",
"face_action", "environment",)
FUNCTION = "process"
@@ -59,7 +61,7 @@ class CSVPromptsLoader():
df = pd.read_csv(path, index_col=0, dtype={
"name": 'string', "styles": 'string', "face": 'string', "cloth": 'string', "action": 'string', "face_action": 'string', "environment": 'string'})
df.fillna("")
df = df.fillna("")
if remove_extension == "On":
key = os.path.splitext(key)[0]