Deeokay commited on
Commit
220c462
1 Parent(s): 0f7da9a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -45,7 +45,7 @@ This is the model card of a 🤗 transformers model that has been pushed on the
45
  ```python
46
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
47
 
48
- models_folder = "Deeokay/gpt2-ownData2"
49
 
50
  model = GPT2LMHeadModel.from_pretrained(models_folder)
51
  tokenizer = GPT2Tokenizer.from_pretrained(models_folder)
@@ -60,10 +60,10 @@ set_seed(42)
60
 
61
  sample_output = model.generate(
62
  **inputs,
63
- max_new_tokens=150,
64
  do_sample=True,
65
- top_k=50,
66
- temperature=0.6,
67
  )
68
 
69
  print("Output:\n" + 80 * '-')
 
45
  ```python
46
  from transformers import GPT2LMHeadModel, GPT2Tokenizer
47
 
48
+ models_folder = "Deeokay/gpt2-javis-stks"
49
 
50
  model = GPT2LMHeadModel.from_pretrained(models_folder)
51
  tokenizer = GPT2Tokenizer.from_pretrained(models_folder)
 
60
 
61
  sample_output = model.generate(
62
  **inputs,
63
+ max_new_tokens=250,
64
  do_sample=True,
65
+ top_k=30,
66
+ temperature=0.7,
67
  )
68
 
69
  print("Output:\n" + 80 * '-')