let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

auto switch mps device to cpu device

Browse Source
This commit is contained in:
Qing
2023-02-11 13:30:09 +08:00
parent f9b5dcbfd7
commit 8f8bcfe0f4
15 changed files with 52 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lama_cleaner/model/manga.py
View File

@@ -76,6 +76,7 @@ MANGA_LINE_MODEL_URL = os.environ.get(
class Manga(InpaintModel):
name = "manga"
pad_mod = 16
def init_model(self, device, **kwargs):