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 CHANGED
@@ -12,7 +12,7 @@ tags:
12
  - time-series
13
  - electricity
14
  - zero-shot
15
- suggested_hardware: t4-small
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 = 256,
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 = 256,
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
  """