omkar56 commited on
Commit
704842e
1 Parent(s): ee4dd0e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -41,7 +41,7 @@ def generate_text(
41
  repetition_penalty = body.get("repetition_penalty", 1.0)
42
  print(f"temperature + {temperature}")
43
  history = [] # You might need to handle this based on your actual usage
44
- formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
45
 
46
  stream = client.text_generation(
47
  formatted_prompt,
 
41
  repetition_penalty = body.get("repetition_penalty", 1.0)
42
  print(f"temperature + {temperature}")
43
  history = [] # You might need to handle this based on your actual usage
44
+ formatted_prompt = format_prompt(f"{sys_prompt}, {prompt}", history)
45
 
46
  stream = client.text_generation(
47
  formatted_prompt,