Update lama-cleaner dep to iopaint

mallman · May 24, 2024 · 557e6de · 557e6de
1 parent 07d22de
commit 557e6de
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/convert_lama.py b/convert_lama.py
@@ -1,7 +1,7 @@
 import coremltools as ct
 import torch
 
-from lama_cleaner.model.lama import LaMa
+from iopaint.model.lama import LaMa
 from CoreMLaMa import CoreMLaMa
 
 model_manager = LaMa("cpu")

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,3 @@
 torch==2.2.0
-lama-cleaner==1.2.5
+iopaint==1.3.3
 coremltools==7.2