---
title: InternViT Development Test
emoji: 🔧
colorFrom: indigo
colorTo: purple
sdk: docker
pinned: false
---

# InternViT-6B with CUDA Development Tools

This Space uses the PyTorch CUDA development image to properly install flash-attn with NVCC.

## Changes in this Version

- Using PyTorch CUDA development image instead of runtime image
- Includes NVCC (NVIDIA CUDA Compiler) needed for flash-attn
- Specific flash-attn version (1.0.9) compatible with CUDA 11.7
- Enhanced diagnostics to verify flash-attn installation

## Dependencies Added

- einops: Required for vision transformer operations
- flash-attn: Required for efficient attention computation
- CUDA build tools for proper compilation

## Instructions

1. Click the "Test Model Loading" button
2. Wait for the model to load and run the test
3. Check the results for success or errors