Upload 2 files

post_install.py

@@ -74,6 +74,11 @@ def install_torch_sparse():
             print("Will try to work with current PyTorch version and attempt alternative installation methods")
             PYTORCH_VERSION = pytorch_base
             CUDA_VERSION = cuda_version
+            
+            # Install torchvision for the current PyTorch version
+            print("Installing torchvision for current PyTorch version...")
+            if not run_command("pip install torchvision"):
+                print("Warning: Failed to install torchvision")
         else:
             # Store version info for later use
             PYTORCH_VERSION = pytorch_base
@@ -385,6 +390,21 @@ def install_pytorch3d():
     
     return False
 
+def install_pytorch_dependencies():
+    """Install PyTorch-related dependencies"""
+    print("Installing PyTorch dependencies...")
+    
+    # Install torchvision
+    if not run_command("pip install torchvision"):
+        print("Warning: Failed to install torchvision")
+        return False
+    
+    # Install torchaudio if needed
+    if not run_command("pip install torchaudio"):
+        print("Warning: Failed to install torchaudio")
+    
+    return True
+
 def install_fashion_clip():
     """Setup Fashion-CLIP"""
     print("Setting up Fashion-CLIP...")
@@ -431,6 +451,10 @@ def main():
         print("Failed to install Fashion-CLIP")
         sys.exit(1)
     
+    if not install_pytorch_dependencies():
+        print("Failed to install PyTorch dependencies")
+        sys.exit(1)
+    
     # Final verification
     print("\n=== Final Verification ===")
     print("Checking all dependencies...")

requirements.txt

@@ -1,4 +1,5 @@
 # Core dependencies - PyTorch will be installed by post_install.py with correct CUDA version
+torchvision
 clip
 imageio
 cython