()
| 78 | sys.exit(1) |
| 79 | |
| 80 | def main(): |
| 81 | print("Starting model downloads...") |
| 82 | download_sam_model() |
| 83 | download_minimax_models() |
| 84 | print("\nAll models downloaded successfully!") |
| 85 | print("\nProject structure should now look like this:") |
| 86 | print(""" |
| 87 | MiniMax-Remover/ |
| 88 | ├── sam_vit_h_4b8939.pth |
| 89 | ├── models/ |
| 90 | │ ├── minimax_vae/ (VAE encoder/decoder) |
| 91 | │ ├── minimax_transformer/ (main diffusion model) |
| 92 | │ └── minimax_scheduler/ (denoising scheduler) |
| 93 | ├── minimax_mask_node_bmo.py |
| 94 | └── pipeline_minimax_remover_bmo.py |
| 95 | """) |
| 96 | |
| 97 | if __name__ == "__main__": |
| 98 | main() |
no test coverage detected