style refine

2022-04-04 21:51:33 +08:00
parent 855fd1f006
commit 10b35a3f0a
33 changed files with 263 additions and 686 deletions
--- a/lama_cleaner/lama/init.py
+++ b/lama_cleaner/lama/init.py
@@ -35,6 +35,7 @@ class LaMa:
        else:
            model_path = download_model(LAMA_MODEL_URL)

+        print(f"Load LaMa model from: {model_path}")
        model = torch.jit.load(model_path, map_location="cpu")
        model = model.to(device)
        model.eval()