Commit 48d096f7 authored Jul 20, 2022 by dmMaze

hip backend for amd GPU

parent ef543b3f

ballontranslator/dl/inpaint/init.py

+2 −2

Original line number	Diff line number	Diff line
		@@ -113,7 +113,7 @@ class AOTInpainter(InpainterBase):
		'options': [
		'cpu',
		'cuda',
		'dml'
		'hip'
		],
		'select': DEFAULT_DEVICE
		},
		@@ -219,7 +219,7 @@ class LamaInpainterMPE(InpainterBase):
		'options': [
		'cpu',
		'cuda',
		'dml'
		'hip'
		],
		'select': DEFAULT_DEVICE
		}

ballontranslator/dl/ocr/init.py

+3 −3

Original line number	Diff line number	Diff line
		@@ -64,7 +64,7 @@ class OCRMIT32px(OCRBase):
		'options': [
		'cpu',
		'cuda',
		'dml'
		'hip'
		],
		'select': DEFAULT_DEVICE
		},
		@@ -117,7 +117,7 @@ class MangaOCR(OCRBase):
		'options': [
		'cpu',
		'cuda',
		'dml'
		'hip'
		],
		'select': DEFAULT_DEVICE
		}
		@@ -177,7 +177,7 @@ class OCRMIT48pxCTC(OCRBase):
		'options': [
		'cpu',
		'cuda',
		'dml'
		'hip'
		],
		'select': DEFAULT_DEVICE
		},

ballontranslator/dl/textdetector/init.py

+1 −1

Original line number	Diff line number	Diff line
		@@ -55,7 +55,7 @@ class ComicTextDetector(TextDetectorBase):
		'options': [
		'cpu',
		'cuda',
		'dml'
		'hip'
		],
		'select': DEFAULT_DEVICE
		},