lots of update 2

2023-01-07 08:52:11 +08:00
parent a22536becc
commit a7240eedb5
13 changed files with 302 additions and 71 deletions
--- a/lama_cleaner/model/sd.py
+++ b/lama_cleaner/model/sd.py
@@ -62,6 +62,7 @@ class SD(InpaintModel):

        if kwargs.get('cpu_offload', False) and torch.cuda.is_available():
            # TODO: gpu_id
+            logger.info("Enable sequential cpu offload")
            self.model.enable_sequential_cpu_offload(gpu_id=0)
        else:
            if kwargs['sd_cpu_textencoder']: