lots of update 2

lama_cleaner/tests/test_paint_by_example.py

@@ -78,3 +78,18 @@ def test_paint_by_example_cpu_offload(strategy):
         fy=0.9,
         fx=1.3
     )
+
+
+@pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL])
+def test_paint_by_example_cpu_offload_cpu_device(strategy):
+    model = ModelManager(name="paint_by_example", device = torch.device('cpu'), cpu_offload=True)
+    cfg = get_config(strategy, paint_by_example_steps=1, sd_scale=0.85)
+    assert_equal(
+        model,
+        cfg,
+        f"paint_by_example_{strategy.capitalize()}_cpu_offload_cpu_device.png",
+        img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
+        mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
+        fy=0.9,
+        fx=1.3
+    )