Upload README.md
Browse files
README.md
CHANGED
|
@@ -39,7 +39,7 @@ from transformers import AutoModelForCausalLM, AutoConfig
|
|
| 39 |
import torch
|
| 40 |
|
| 41 |
# Load the model
|
| 42 |
-
model = AutoModelForCausalLM.from_pretrained("
|
| 43 |
|
| 44 |
# Prepare input time series data
|
| 45 |
# Shape: [batch_size, sequence_length, features]
|
|
@@ -62,15 +62,19 @@ Key configuration parameters:
|
|
| 62 |
- `freq_experts`: Frequency-specific expert configuration
|
| 63 |
- `moe_temp`: Temperature for expert selection during inference (default: 1)
|
| 64 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
## Citation
|
| 66 |
|
| 67 |
If you use SuperLinear in your research, please cite:
|
| 68 |
|
| 69 |
```bibtex
|
| 70 |
-
@article{
|
| 71 |
-
title={SuperLinear:
|
| 72 |
author={Your Name},
|
| 73 |
-
year={
|
| 74 |
}
|
| 75 |
```
|
| 76 |
|
|
|
|
| 39 |
import torch
|
| 40 |
|
| 41 |
# Load the model
|
| 42 |
+
model = AutoModelForCausalLM.from_pretrained("SequentialLearning/SuperLinear", trust_remote_code=True)
|
| 43 |
|
| 44 |
# Prepare input time series data
|
| 45 |
# Shape: [batch_size, sequence_length, features]
|
|
|
|
| 62 |
- `freq_experts`: Frequency-specific expert configuration
|
| 63 |
- `moe_temp`: Temperature for expert selection during inference (default: 1)
|
| 64 |
|
| 65 |
+
## Link to GitHub
|
| 66 |
+
|
| 67 |
+
[https://github.com/azencot-group/SuperLinear](https://github.com/azencot-group/SuperLinear)
|
| 68 |
+
|
| 69 |
## Citation
|
| 70 |
|
| 71 |
If you use SuperLinear in your research, please cite:
|
| 72 |
|
| 73 |
```bibtex
|
| 74 |
+
@article{todo,
|
| 75 |
+
title={SuperLinear: todo},
|
| 76 |
author={Your Name},
|
| 77 |
+
year={2025}
|
| 78 |
}
|
| 79 |
```
|
| 80 |
|