kyujinpy commited on
Commit
fb3f55e
1 Parent(s): 60e0201

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -115,7 +115,7 @@ I use A100 GPU 40GB and COLAB, when trianing.
115
  from transformers import AutoModelForCausalLM, AutoTokenizer
116
  import torch
117
 
118
- repo = "MarkrAI/kyujinpy-KoT-platypus2-7B"
119
  CoT-llama = AutoModelForCausalLM.from_pretrained(
120
  repo,
121
  return_dict=True,
 
115
  from transformers import AutoModelForCausalLM, AutoTokenizer
116
  import torch
117
 
118
+ repo = "kyujinpy/KoT-platypus2-7B"
119
  CoT-llama = AutoModelForCausalLM.from_pretrained(
120
  repo,
121
  return_dict=True,