skylersterling commited on
Commit
ff90fb4
·
verified ·
1 Parent(s): 403b4a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def generate_text(prompt, temperature, top_p):
23
  prompt_length = len(generated_text)
24
 
25
  if input_tokens.size(1) > 512:
26
- generated = "ERROR, CONTEXT SIZE EXCEEDED"
27
 
28
  for _ in range(80): # Adjust the range to control the number of tokens generated
29
  with torch.no_grad():
 
23
  prompt_length = len(generated_text)
24
 
25
  if input_tokens.size(1) > 512:
26
+ generated_text = "ERROR, CONTEXT SIZE EXCEEDED"
27
 
28
  for _ in range(80): # Adjust the range to control the number of tokens generated
29
  with torch.no_grad():