title: InternViT Fixed BFloat16 Type | |
emoji: π | |
colorFrom: green | |
colorTo: blue | |
sdk: docker | |
pinned: false | |
# InternViT-6B with Fixed BFloat16 Type Conversion | |
This Space demonstrates loading and using the InternViT-6B model with proper BFloat16 type handling. | |
## Fixes Applied | |
1. Fixed the data type mismatch between input tensors (float32) and model parameters (bfloat16) | |
2. Always converts input tensors to bfloat16 before processing | |
3. Uses a dummy flash-attn implementation to avoid compilation | |
4. Provides detailed diagnostics on startup | |
## Benefits | |
- Much faster build time | |
- No data type mismatches or conversion errors | |
- Still loads the full 6B parameter model | |
- Demonstrates basic functionality | |
## 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 | |