File size: 854 Bytes
49cb844 56d4429 49cb844 56d4429 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
---
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
|