davidkim205 commited on
Commit
e7a7dce
·
verified ·
1 Parent(s): bea716a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -8
README.md CHANGED
@@ -28,7 +28,6 @@ Iris provides efficient and accurate translation and can be used in a variety of
28
  ## usage
29
  ```
30
  from transformers import AutoModelForCausalLM, AutoTokenizer
31
- from utils.simple_bleu import simple_score
32
  import torch
33
 
34
  repo = "davidkim205/iris-7b"
@@ -70,20 +69,19 @@ def main():
70
  ko_text = translate_en2ko(en_text)
71
  print('en_text', en_text)
72
  print('ko_text', ko_text)
73
- print('score', simple_score(text, ko_text))
74
 
75
  if __name__ == "__main__":
76
  main()
77
  ```
78
  output
 
79
  ```
80
  $ python iris_test.py
81
- Loading checkpoint shards: 100%|██████████████████████████████████████████████████████████████████████████████| 3/3 [00:02<00:00, 1.05it/s]
82
- >Iris is a model for Korean-English sentence translation based on deep learning.
83
- Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.
84
- Setting `pad_token_id` to `eos_token_id`:2 for open-end generation.
85
- en_text Iris is a model for Korean-English sentence translation based on deep learning.</s>
86
- ko_text 아이리스는 딥러닝을 기반으로 한 한-영어 문장 번역을 위한 모델이다.</s>
87
 
88
  ```
89
  ## template
 
28
  ## usage
29
  ```
30
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
31
  import torch
32
 
33
  repo = "davidkim205/iris-7b"
 
69
  ko_text = translate_en2ko(en_text)
70
  print('en_text', en_text)
71
  print('ko_text', ko_text)
 
72
 
73
  if __name__ == "__main__":
74
  main()
75
  ```
76
  output
77
+
78
  ```
79
  $ python iris_test.py
80
+ Downloading shards: 100%|█████████████████████████████████████████████████████████████████████████████████████| 3/3 [00:00<00:00, 4.72it/s]
81
+ Loading checkpoint shards: 100%|██████████████████████████████████████████████████████████████████████████████| 3/3 [00:02<00:00, 1.07it/s]
82
+ >아이리스는 딥러닝을 기반으로 한-영어 문장 번역을 위한 모델이다.
83
+ en_text Iris is a model for Korean-to-English sentence translation based on deep learning.</s>
84
+ ko_text 아이리스는 딥러닝을 기반으로 한국어-영어 문장 번역을 위한 모델이다.</s>
 
85
 
86
  ```
87
  ## template