Update lama_cleaner/model/lama.py

lama_cleaner/model/lama.py

@@ -19,16 +19,16 @@ class LaMa(InpaintModel):
     pad_mod = 8
 
     def init_model(self, device, **kwargs):
-        if os.environ.get("LAMA_MODEL"):
-            model_path = os.environ.get("LAMA_MODEL")
-            if not os.path.exists(model_path):
-                raise FileNotFoundError(
-                    f"lama torchscript model not found: {model_path}"
-                )
-        else:
-            model_path = download_model(LAMA_MODEL_URL)
-        # TODO used to create a lambda docker image
-        # model_path = '../app/big-lama.pt'
+        # if os.environ.get("LAMA_MODEL"):
+        #     model_path = os.environ.get("LAMA_MODEL")
+        #     if not os.path.exists(model_path):
+        #         raise FileNotFoundError(
+        #             f"lama torchscript model not found: {model_path}"
+        #         )
+        # else:
+        #     # model_path = download_model(LAMA_MODEL_URL)
+        #     # TODO used to create a lambda docker image
+        model_path = 'app/big-lama.pt'
         logger.info(f"Load LaMa model from: {model_path}")
         model = torch.jit.load(model_path, map_location="cpu")
         model = model.to(device)