timesfm icon indicating copy to clipboard operation
timesfm copied to clipboard

Published 20 hours ago • verified publisher icon google-research

Jitting fails with cpu, gpu & cuda

Open Blochergroup opened this issue 9 months ago • 3 comments

Tried running this library: https://github.com/gabaid971/ts-forecaster

Doesn't matter If I try with my CPU, GPU (radeon 780 integrated), or cuda (RTX 4060). When i run it in a jupyter notebook it crashes during "jitting decoding" when I load from checkpoint

My parameters: tfm = timesfm.TimesFm( context_len=128, horizon_len=14, input_patch_len=32, output_patch_len=128, num_layers=20, model_dims=1280, backend="cpu", )

Blochergroup avatar May 24 '24 09:05 Blochergroup