Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"
|