Upload README.md
Browse files
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 = "
|
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,
|