nodes/diontimmer-ComfyUI-Vextra-Nodes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix paths for promptgen

Browse Source
This commit is contained in:
Dion Timmer
2023-04-06 21:48:47 -04:00
parent eb30f44ec7
commit e73a6d5765
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes/DT_PromptGen.py
+2 -2
View File
@@ -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():
"""