Skip to content

Commit a9c8a6d

Browse files
committed
fix: use torch_dtype for PaddleOCR-VL model loading
1 parent a809946 commit a9c8a6d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

manga_translator/ocr/model_paddleocr_vl.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,7 @@ async def _load(self, device: str):
254254
self.model = AutoModel.from_pretrained(
255255
load_path,
256256
trust_remote_code=True,
257-
dtype=model_dtype,
257+
torch_dtype=model_dtype,
258258
device_map=self.device if self.device != 'cpu' else None,
259259
local_files_only=True
260260
)

0 commit comments

Comments
 (0)