gokaygokay commited on
Commit
67d1e7e
1 Parent(s): 868c4e3

prompt_types

Browse files
Files changed (1) hide show
  1. huggingface_inference_node.py +5 -1
huggingface_inference_node.py CHANGED
@@ -64,8 +64,12 @@ You are allowed to make up film and branding names, and do them like 80's, 90's
64
 
65
  if custom_base_prompt.strip():
66
  base_prompt = custom_base_prompt
 
 
 
67
  else:
68
- base_prompt = prompt_types.get(prompt_type, default_happy_prompt)
 
69
 
70
  if compress and not poster:
71
  compression_chars = {
 
64
 
65
  if custom_base_prompt.strip():
66
  base_prompt = custom_base_prompt
67
+ elif prompt_type in prompt_types:
68
+ base_prompt = prompt_types[prompt_type]
69
+ print(f"Using {prompt_type} prompt")
70
  else:
71
+ base_prompt = default_happy_prompt
72
+ print(f"Warning: Unknown prompt type '{prompt_type}'. Using default happy prompt.")
73
 
74
  if compress and not poster:
75
  compression_chars = {