skylersterling commited on
Commit
645e147
·
verified ·
1 Parent(s): 8408816

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -17,6 +17,8 @@ def generate_text(prompt, temperature=0.1, top_p=0.2, max_tokens=3):
17
  prompt_with_eos = prompt + " >" # Add the "EOS" to the end of the prompt
18
  input_tokens = tokenizer.encode(prompt_with_eos, return_tensors='pt')
19
 
 
 
20
  input_tokens = input_tokens.to('cpu')
21
 
22
  generated_text = prompt_with_eos # Start with the initial prompt plus "EOS"
 
17
  prompt_with_eos = prompt + " >" # Add the "EOS" to the end of the prompt
18
  input_tokens = tokenizer.encode(prompt_with_eos, return_tensors='pt')
19
 
20
+ print(prompt_with_eos)
21
+
22
  input_tokens = input_tokens.to('cpu')
23
 
24
  generated_text = prompt_with_eos # Start with the initial prompt plus "EOS"