Fix torch==2.5.1, clean up cache files

.gitignore

@@ -1 +1,7 @@
-.DS_Store
+.DS_Store
+__pycache__/
+*.pyc
+.venv*/
+test_local.py
+requirements-local.txt
+setup_local.sh

requirements-local.txt

@@ -1,25 +0,0 @@
-# Local testing requirements (CPU mode, no CUDA)
-numpy<2.0.0
-scikit-learn
-pillow
-torch<2.6
-torchvision
-roma
-matplotlib
-tqdm
-opencv-python
-scipy
-einops
-trimesh
-pyglet<2
-huggingface-hub
-lightning
-hydra-core
-jaxtyping
-beartype
-einops
-colorama
-scikit-image
-dacite
-plyfile
-omegaconf

requirements.txt

@@ -1,7 +1,7 @@
 numpy<2.0.0
 scikit-learn
 pillow-heif
-torch<2.6
+torch==2.5.1
 torchvision
 roma
 gradio>=6.0.0

setup_local.sh

@@ -1,19 +0,0 @@
-#!/bin/bash
-# Setup local testing environment
-
-set -e
-
-echo "Creating virtual environment..."
-python3 -m venv .venv-local
-
-echo "Activating..."
-source .venv-local/bin/activate
-
-echo "Installing dependencies..."
-pip install --upgrade pip
-pip install -r requirements-local.txt
-
-echo ""
-echo "Done! To test:"
-echo "  source .venv-local/bin/activate"
-echo "  python test_local.py"

test_local.py

@@ -1,101 +0,0 @@
-#!/usr/bin/env python3
-"""Local test script to debug model loading without GPU."""
-
-import sys
-import os
-
-# MUST BE FIRST: Fix for PyTorch 2.6+ weights_only default change
-print("1. Importing torch and applying patch...")
-import torch
-_original_torch_load = torch.load
-def _patched_torch_load(*args, **kwargs):
-    kwargs.setdefault('weights_only', False)
-    return _original_torch_load(*args, **kwargs)
-torch.load = _patched_torch_load
-print(f"   torch version: {torch.__version__}")
-print(f"   torch.load patched: {torch.load != _original_torch_load}")
-
-# Force CPU mode
-os.environ['CUDA_VISIBLE_DEVICES'] = ''
-print(f"   CUDA available: {torch.cuda.is_available()}")
-
-# Setup paths
-sys.path.insert(0, 'src/mast3r_src')
-sys.path.insert(0, 'src/mast3r_src/dust3r')
-sys.path.insert(0, 'src/pixelsplat_src')
-
-print("\n2. Testing imports...")
-try:
-    from dust3r.utils.image import load_images
-    print("   dust3r.utils.image: OK")
-except Exception as e:
-    print(f"   dust3r.utils.image: FAILED - {e}")
-
-try:
-    from mast3r.utils.misc import hash_md5
-    print("   mast3r.utils.misc: OK")
-except Exception as e:
-    print(f"   mast3r.utils.misc: FAILED - {e}")
-
-try:
-    import main
-    print("   main: OK")
-except Exception as e:
-    print(f"   main: FAILED - {e}")
-    import traceback
-    traceback.print_exc()
-
-print("\n3. Downloading model weights...")
-try:
-    from huggingface_hub import hf_hub_download
-    model_name = "brandonsmart/splatt3r_v1.0"
-    filename = "epoch=19-step=1200.ckpt"
-    weights_path = hf_hub_download(repo_id=model_name, filename=filename)
-    print(f"   Downloaded: {weights_path}")
-except Exception as e:
-    print(f"   FAILED: {e}")
-    sys.exit(1)
-
-print("\n4. Testing torch.load directly...")
-try:
-    # This should use our patched version
-    ckpt = torch.load(weights_path, map_location='cpu')
-    print(f"   torch.load: OK (keys: {list(ckpt.keys())[:5]}...)")
-except Exception as e:
-    print(f"   torch.load: FAILED - {e}")
-    import traceback
-    traceback.print_exc()
-    sys.exit(1)
-
-print("\n5. Testing model instantiation...")
-try:
-    device = torch.device('cpu')
-    model = main.MAST3RGaussians.load_from_checkpoint(weights_path, device)
-    print(f"   Model loaded: OK")
-    print(f"   Model type: {type(model)}")
-except Exception as e:
-    print(f"   Model load: FAILED - {e}")
-    import traceback
-    traceback.print_exc()
-    sys.exit(1)
-
-print("\n6. Testing image loading...")
-try:
-    # Use a test image if available
-    test_images = []
-    for img in ['../IMG_2043.jpg', '../IMG_2044.jpg']:
-        if os.path.exists(img):
-            test_images.append(img)
-
-    if test_images:
-        imgs = load_images(test_images[:2] if len(test_images) >= 2 else test_images * 2, size=512, verbose=True)
-        print(f"   Images loaded: {len(imgs)}")
-        print(f"   Image shape: {imgs[0]['img'].shape}")
-    else:
-        print("   No test images found, skipping")
-except Exception as e:
-    print(f"   Image load: FAILED - {e}")
-    import traceback
-    traceback.print_exc()
-
-print("\n✓ All tests passed!")