pro-grammer commited on
Commit
0027a44
·
verified ·
1 Parent(s): 9b4550b

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -151,7 +151,7 @@ context = torch.tensor([enc.encode("Once upon a time there was a knight called B
151
 
152
  # Test generation with a higher number of tokens and adjusted temperature
153
  max_new_tokens = 150 # Increase the token limit for a longer generation
154
- temperature = 0.6 # More focused, less random
155
  generated_text_idx = model.generate(context, max_new_tokens)
156
  generated_text = enc.decode(generated_text_idx[0].tolist())
157
 
 
151
 
152
  # Test generation with a higher number of tokens and adjusted temperature
153
  max_new_tokens = 150 # Increase the token limit for a longer generation
154
+ temperature = 0.5 # More focused, less random
155
  generated_text_idx = model.generate(context, max_new_tokens)
156
  generated_text = enc.decode(generated_text_idx[0].tolist())
157