Spaces:
Sleeping
Sleeping
Evgueni Poloukarov
Claude
commited on
Commit
·
42acd7e
1
Parent(s):
dfe9aab
fix: revert context window to 512h and upgrade GPU to 1xL4 (24GB)
Browse files- Reverted context_hours from 256h back to 512h in chronos_inference.py
- Reverted context_hours from 256h back to 512h in dynamic_forecast.py
- Upgraded HF Space hardware from t4-small (16GB) to l4-1x (24GB VRAM)
- 24GB VRAM will handle 12.44 GiB allocation for 615 features x 512h context
- Fixes CUDA Out of Memory error while preserving full lookback period
Co-Authored-By: Claude <[email protected]>
- README.md +1 -1
- src/forecasting/chronos_inference.py +1 -1
- src/forecasting/dynamic_forecast.py +1 -1
README.md
CHANGED
|
@@ -12,7 +12,7 @@ tags:
|
|
| 12 |
- time-series
|
| 13 |
- electricity
|
| 14 |
- zero-shot
|
| 15 |
-
suggested_hardware:
|
| 16 |
suggested_storage: small
|
| 17 |
---
|
| 18 |
|
|
|
|
| 12 |
- time-series
|
| 13 |
- electricity
|
| 14 |
- zero-shot
|
| 15 |
+
suggested_hardware: l4x1
|
| 16 |
suggested_storage: small
|
| 17 |
---
|
| 18 |
|
src/forecasting/chronos_inference.py
CHANGED
|
@@ -105,7 +105,7 @@ class ChronosInferencePipeline:
|
|
| 105 |
run_date: str,
|
| 106 |
borders: Optional[List[str]] = None,
|
| 107 |
forecast_days: int = 7,
|
| 108 |
-
context_hours: int =
|
| 109 |
num_samples: int = 20
|
| 110 |
) -> Dict:
|
| 111 |
"""
|
|
|
|
| 105 |
run_date: str,
|
| 106 |
borders: Optional[List[str]] = None,
|
| 107 |
forecast_days: int = 7,
|
| 108 |
+
context_hours: int = 512,
|
| 109 |
num_samples: int = 20
|
| 110 |
) -> Dict:
|
| 111 |
"""
|
src/forecasting/dynamic_forecast.py
CHANGED
|
@@ -30,7 +30,7 @@ class DynamicForecast:
|
|
| 30 |
def __init__(
|
| 31 |
self,
|
| 32 |
dataset: pl.DataFrame,
|
| 33 |
-
context_hours: int =
|
| 34 |
forecast_hours: int = 336 # Fixed at 14 days
|
| 35 |
):
|
| 36 |
"""
|
|
|
|
| 30 |
def __init__(
|
| 31 |
self,
|
| 32 |
dataset: pl.DataFrame,
|
| 33 |
+
context_hours: int = 512,
|
| 34 |
forecast_hours: int = 336 # Fixed at 14 days
|
| 35 |
):
|
| 36 |
"""
|