How to use GridTST/traffic_96_96 with Transformers:
# Load model directly from transformers import GridTSTForTimeSeriesPrediction model = GridTSTForTimeSeriesPrediction.from_pretrained("GridTST/traffic_96_96", dtype="auto")
