File size: 1,028 Bytes
ecc8338 | 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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | ---
license: mit
language:
- en
tags:
- text-to-sql
- spider
- hrm-text
base_model: sapientai/HRM-Text-1B
---
# HRM-Text-1B-SQL-Spider
Fine-tuned version of [HRM-Text-1B](https://huggingface.co/sapientai/HRM-Text-1B) on the Spider text-to-SQL dataset.
## Model Details
- **Base Model:** sapientai/HRM-Text-1B (1B parameters, hierarchical reasoning model)
- **Training Data:** Spider dataset (~20k examples)
- **Training:** 3 epochs, ~6 minutes on L40S GPU
- **Architecture:** Hierarchical Reasoning Model with H_cycles=2, L_cycles=3
## Performance
| Model | Accuracy |
|-------|----------|
| Base | 8.00% |
| **Fine-tuned** | **70.00%** |
## Usage
## Training Details
- **Framework:** PyTorch with FlashAttention 3
- **Loss:** Cross-entropy
- **Hardware:** AWS L40S GPU
- **Training Time:** ~6 minutes
## Limitations
- Maximum sequence length: 4096 tokens
- Requires FlashAttention 3 for inference (Ada Lovelace or newer GPUs)
- Best performance on Spider-style schema-aware SQL generation
## License
MIT License
|