storytold/storyteller-ml
2
0
Fork 0
mirror of https://github.com/storytold/storyteller-ml.git synced 2026-10-17 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

make prompt path configurable

Browse Source
This commit is contained in:
Brandon Thomas
2024-02-04 01:33:12 -05:00
parent 93e84f452d
commit 626c2062cb
1 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
workflows/comfy/ComfyLauncher/ComfyRunner.py
+16 -2
View File
@@ -7,8 +7,22 @@ import signal
import aiohttp
current_script_dir = os.path.dirname(os.path.realpath(__file__))
file_path = os.path.join(current_script_dir, 'prompt', 'prompt.json')
import argparse
def parse_args():
parser = argparse.ArgumentParser(description='Run Comfy')
parser.add_argument('--prompt', type=str, help='location of the prompt json', required=True)
return parser.parse_args()
# current_script_dir = os.path.dirname(os.path.realpath(__file__))
# file_path = os.path.join(current_script_dir, 'prompt', 'prompt.json')
args = parse_args()
file_path = args.prompt
print(f'Loading prompt json from path: {file_path}')
prompt = json.load(open(file_path))
comfy_path = os.path.dirname(os.path.abspath(__file__))