From fdffc28271b02699aeef997e5c557e859abd3bae Mon Sep 17 00:00:00 2001
From: casterpollux <brendan@casterpollux.com>
Date: Mon, 16 Jun 2025 21:21:20 -0700
Subject: [PATCH] Delete test_minimax_remover.py

---
 test_minimax_remover.py | 56 -----------------------------------------
 1 file changed, 56 deletions(-)
 delete mode 100644 test_minimax_remover.py

diff --git a/test_minimax_remover.py b/test_minimax_remover.py
deleted file mode 100644
index a2b163c..0000000
--- a/test_minimax_remover.py
+++ /dev/null
@@ -1,56 +0,0 @@
-import torch
-from diffusers.utils import export_to_video
-from decord import VideoReader
-from diffusers.models import AutoencoderKLWan
-from transformer_minimax_remover import Transformer3DModel
-from diffusers.schedulers import UniPCMultistepScheduler
-from pipeline_minimax_remover import Minimax_Remover_Pipeline
-
-random_seed = 42
-video_length = 81
-device = torch.device("cuda:0")
-
-vae = AutoencoderKLWan.from_pretrained("./vae", torch_dtype=torch.float16)
-transformer = Transformer3DModel.from_pretrained("./transformer", torch_dtype=torch.float16)
-scheduler = UniPCMultistepScheduler.from_pretrained("./scheduler")
-
-pipe = Minimax_Remover_Pipeline(transformer=transformer, vae=vae, scheduler=scheduler)
-pipe.to(device)
-
-# the iterations is the hyperparameter for mask dilation
-def inference(pixel_values, masks, iterations=6):
-    video = pipe(
-        images=pixel_values,
-        masks=masks,
-        num_frames=video_length,
-        height=480,
-        width=832,
-        num_inference_steps=12,
-        generator=torch.Generator(device=device).manual_seed(random_seed),
-        iterations=iterations
-    ).frames[0]
-    export_to_video(video, "./output.mp4")
-
-def load_video(video_path):
-    vr = VideoReader(video_path)
-    images = vr.get_batch(list(range(video_length))).asnumpy()
-    images = torch.from_numpy(images)/127.5 - 1.0
-    return images
-
-def load_mask(mask_path):
-    vr = VideoReader(mask_path)
-    masks = vr.get_batch(list(range(video_length))).asnumpy()
-    masks = torch.from_numpy(masks)
-    masks = masks[:, :, :, :1]
-    masks[masks > 20] = 255
-    masks[masks < 255] = 0
-    masks = masks / 255.0
-    return masks
-
-video_path = "./video.mp4"
-mask_path = "./mask.mp4"
-
-images = load_video(video_path)
-masks = load_mask(mask_path)
-
-inference(images, masks)
\ No newline at end of file