idah4 commited on
Commit
a12f029
·
verified ·
1 Parent(s): 180d231

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -34,7 +34,7 @@ import torch
34
  from transformers import AutoModelForCausalLM
35
 
36
  # 1️⃣ 모델 로드
37
- repo_id = "idah4/byteetm-korean"
38
  device = "cuda" if torch.cuda.is_available() else "cpu"
39
 
40
  model = AutoModelForCausalLM.from_pretrained(
 
34
  from transformers import AutoModelForCausalLM
35
 
36
  # 1️⃣ 모델 로드
37
+ repo_id = "idah4/byteetm-korean-tiny"
38
  device = "cuda" if torch.cuda.is_available() else "cpu"
39
 
40
  model = AutoModelForCausalLM.from_pretrained(