Skip to content

Commit

Permalink
change temperature
Browse files Browse the repository at this point in the history
  • Loading branch information
taprosoft committed Aug 17, 2023
1 parent eb43843 commit e526b24
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nanoGPT/sample.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
start = "\n" # or "<|endoftext|>" or etc. Can also specify a file, use as: "FILE:prompt.txt"
num_samples = 10 # number of samples to draw
max_new_tokens = 500 # number of tokens generated in each sample
temperature = 0.8 # 1.0 = no change, < 1.0 = less random, > 1.0 = more random, in predictions
temperature = 0.0 # 1.0 = no change, < 1.0 = less random, > 1.0 = more random, in predictions
top_k = 200 # retain only the top_k most likely tokens, clamp others to have 0 probability
seed = 1337
device = 'cuda' # examples: 'cpu', 'cuda', 'cuda:0', 'cuda:1', etc.
Expand Down

0 comments on commit e526b24

Please sign in to comment.