yujuanqin commited on
Commit
69f7e51
·
1 Parent(s): b295d06
Files changed (2) hide show
  1. lib/models/mac/llm.py +2 -2
  2. lib/utils.py +1 -1
lib/models/mac/llm.py CHANGED
@@ -70,10 +70,10 @@ class QwenTranslator:
70
  ]
71
 
72
  def _warmup(self):
73
- self.translate(prompt="hello", src_lang="en", dst_lang="zh")
74
 
75
  @lru_cache(maxsize=10)
76
- def translate(self, prompt, src_lang, dst_lang) -> str:
77
  message = self.to_message(prompt, src_lang, dst_lang)
78
  with Timer("llm inference") as t:
79
  output = self.llm.create_chat_completion(messages=message, temperature=0)
 
70
  ]
71
 
72
  def _warmup(self):
73
+ self.translate(prompt="how are you?", src_lang="en", dst_lang="zh")
74
 
75
  @lru_cache(maxsize=10)
76
+ def translate(self, prompt, src_lang, dst_lang):
77
  message = self.to_message(prompt, src_lang, dst_lang)
78
  with Timer("llm inference") as t:
79
  output = self.llm.create_chat_completion(messages=message, temperature=0)
lib/utils.py CHANGED
@@ -48,7 +48,7 @@ def get_time_str(level:Literal["d","s","ms"]="d"):
48
 
49
 
50
  def save_csv(file_path, header, rows):
51
- with open(file_path, "w", encoding="utf-8") as f:
52
  writer = csv.writer(f)
53
  if header:
54
  writer.writerow(header)
 
48
 
49
 
50
  def save_csv(file_path, header, rows):
51
+ with open(file_path, "w", encoding="utf-8", newline="") as f:
52
  writer = csv.writer(f)
53
  if header:
54
  writer.writerow(header)