no longer return extra images for classic relight + formatting

zsxkib · May 21, 2024 · debe04e · debe04e
1 parent 541e6ed
commit debe04e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/predict.py b/predict.py
@@ -311,7 +311,7 @@ def process_relight(
         bg_source,
     )
     results = [(x * 255.0).clip(0, 255).astype(np.uint8) for x in results]
-    return results + extra_images
+    return results  # + extra_images
 
 
 @torch.inference_mode()
@@ -839,4 +839,4 @@ def predict(
             Image.fromarray(img).save(img_path, **save_params)
             output_paths.append(Path(img_path))
 
-        return output_paths
+        return output_paths[::-1]