From e73a6d576581e35a88b219ec991703cb2a0ca293 Mon Sep 17 00:00:00 2001 From: Dion Timmer Date: Thu, 6 Apr 2023 21:48:47 -0400 Subject: [PATCH] fix paths for promptgen --- nodes/DT_PromptGen.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nodes/DT_PromptGen.py b/nodes/DT_PromptGen.py index eda19ad..7d1378d 100644 --- a/nodes/DT_PromptGen.py +++ b/nodes/DT_PromptGen.py @@ -5,13 +5,13 @@ import random script_path = os.path.dirname(os.path.realpath(__file__)) comfy_path = script_path.split('custom_nodes')[0] -if not os.path.exists(f'{comfy_path}custom_nodes/VextraNodes/binary/distilgpt2-stable-diffusion-v2'): +if not os.path.exists(f'{comfy_path}custom_nodes/ComfyUI-Vextra-Nodes/binary/distilgpt2-stable-diffusion-v2'): print('Downloading model...') os.system(f'git clone https://huggingface.co/FredZhang7/distilgpt2-stable-diffusion-v2 {comfy_path}custom_nodes/ComfyUI-Vextra-Nodes/binary/distilgpt2-stable-diffusion-v2') tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2') tokenizer.add_special_tokens({'pad_token': '[PAD]'}) -model = GPT2LMHeadModel.from_pretrained(f'{comfy_path}custom_nodes/VextraNodes/binary/distilgpt2-stable-diffusion-v2') +model = GPT2LMHeadModel.from_pretrained(f'{comfy_path}custom_nodes/ComfyUI-Vextra-Nodes/binary/distilgpt2-stable-diffusion-v2') class GetPrompt(): """