Update cuda in pythainlp.translate

wannaphong · wannaphong · commit 2e7dc23c9ce6 · 2023-07-03T02:54:16.000+07:00
diff --git a/pythainlp/translate/en_th.py b/pythainlp/translate/en_th.py
@@ -87,7 +87,7 @@ def __init__(self, use_gpu: bool = False):
             ),
         )
         if use_gpu:
-            self._model.cuda()
+            self._model = self._model.cuda()
 
     def translate(self, text: str) -> str:
         """
diff --git a/pythainlp/translate/th_fr.py b/pythainlp/translate/th_fr.py
@@ -49,7 +49,7 @@ def __init__(
         self.tokenizer_thzh = AutoTokenizer.from_pretrained(pretrained)
         self.model_thzh = AutoModelForSeq2SeqLM.from_pretrained(pretrained)
         if use_gpu:
-            self.model_thzh.cuda()
+            self.model_thzh = self.model_thzh.cuda()
 
     def translate(self, text: str) -> str:
         """
diff --git a/pythainlp/translate/zh_th.py b/pythainlp/translate/zh_th.py
@@ -43,7 +43,7 @@ def __init__(
         self.tokenizer_thzh = AutoTokenizer.from_pretrained(pretrained)
         self.model_thzh = AutoModelForSeq2SeqLM.from_pretrained(pretrained)
         if use_gpu:
-            self.model_thzh.cuda()
+            self.model_thzh = self.model_thzh.cuda()
 
     def translate(self, text: str) -> str:
         """

Original file line number	Diff line number	Diff line change
`@@ -87,7 +87,7 @@ def __init__(self, use_gpu: bool = False):`
`87`	`87`	`),`
`88`	`88`	`)`
`89`	`89`	`if use_gpu:`
`90`		`- self._model.cuda()`
	`90`	`+ self._model = self._model.cuda()`
`91`	`91`
`92`	`92`	`def translate(self, text: str) -> str:`
`93`	`93`	`"""`