ysd990 commited on
Commit
faf5606
·
verified ·
1 Parent(s): b1bff7d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -21,7 +21,7 @@ import os
21
  import torch
22
  from transformers import AutoTokenizer, AutoModelForCausalLM
23
 
24
- model_id = 'KAERI-MLP/gemma2-Korean-AtomicGPT-9B'
25
 
26
  tokenizer = AutoTokenizer.from_pretrained(model_id)
27
  model = AutoModelForCausalLM.from_pretrained(
 
21
  import torch
22
  from transformers import AutoTokenizer, AutoModelForCausalLM
23
 
24
+ model_id = 'KAERI-MLP/AtomicGPT-gemma2-9B'
25
 
26
  tokenizer = AutoTokenizer.from_pretrained(model_id)
27
  model = AutoModelForCausalLM.from_pretrained(