18337 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Jinan Zhou	8deeb3e191	fix comments	2025-02-28 11:25:51 -08:00
Kashif Rasul	380e6bff37	add _CHECKPOINT_FOR_DOC	2025-02-28 09:06:18 +01:00
Kashif Rasul	909fd6c6f4	Merge branch 'timesfm' of https://github.com/kashif/transformers into timesfm	2025-02-28 07:30:10 +01:00
Kashif Rasul	0026ba6ff0	add docstrings to output data classes	2025-02-27 16:00:58 -08:00
Kashif Rasul	f76116bc65	fix logging	2025-02-27 16:00:58 -08:00
Kashif Rasul	b7e75e9f74	add doc strings	2025-02-27 16:00:58 -08:00
Kashif Rasul	71bda445b7	override generate as its not needed	2025-02-27 16:00:58 -08:00
Kashif Rasul	c0a4f48759	fix docstring	2025-02-27 16:00:58 -08:00
Kashif Rasul	df7bbb0610	ignore TimesFmModel for testing	2025-02-27 16:00:58 -08:00
Kashif Rasul	4466315657	fix formatting	2025-02-27 16:00:58 -08:00
Kashif Rasul	02e62c6dc9	use TimesFmModel	2025-02-27 16:00:58 -08:00
Kashif Rasul	913f3608be	2.0-500m-pytorch default configs	2025-02-27 16:00:58 -08:00
Kashif Rasul	f460370e53	TimesFmPositionalEmbedding takes config for its init	2025-02-27 16:00:58 -08:00
Kashif Rasul	5352cda52f	TimesFmModel instead of TimesFmDecoder	2025-02-27 16:00:58 -08:00
Kashif Rasul	a75b8e7a2d	use initializer_range	2025-02-27 16:00:58 -08:00
Kashif Rasul	d8c2e0d74f	fix tests	2025-02-27 16:00:58 -08:00
Kashif Rasul	ef59621e80	use T5LayerNorm	2025-02-27 16:00:58 -08:00
Kashif Rasul	038859d616	renaem TimesFmMLP	2025-02-27 16:00:58 -08:00
Kashif Rasul	ea05e276c1	simpler init	2025-02-27 16:00:58 -08:00
Kashif Rasul	b428972d4f	fix-copies	2025-02-27 16:00:58 -08:00
Jinan Zhou	c65e4b46db	more readable code for weight conversion	2025-02-27 16:00:58 -08:00
Jinan Zhou	53b290aef7	consolidate decode func	2025-02-27 16:00:58 -08:00
Jinan Zhou	22bb7cfad0	refactor type hints, use CamelCase	2025-02-27 16:00:58 -08:00
Kashif Rasul	7c7e56f8f4	compare mean forecasts	2025-02-27 16:00:58 -08:00
Rajat Sen	cef8510317	bug fix in timesfm decode function.	2025-02-27 16:00:58 -08:00
Kashif Rasul	91c50a4aaf	use timesfm 2.0	2025-02-27 16:00:58 -08:00
Kashif Rasul	686c71bb93	add doc for SDPA	2025-02-27 16:00:58 -08:00
Kashif Rasul	abb1c0ae63	fix _quantile_loss	2025-02-27 16:00:58 -08:00
Kashif Rasul	689d2a4514	fix import order	2025-02-27 16:00:58 -08:00
Kashif Rasul	c46864435a	fix formatting	2025-02-27 16:00:58 -08:00
Kashif Rasul	be8922fa97	added initial TimesFMModelIntegrationTests	2025-02-27 16:00:58 -08:00
Kashif Rasul	9aad1013f0	fix quantile loss	2025-02-27 16:00:58 -08:00
Kashif Rasul	b41c3686a7	fix file name	2025-02-27 16:00:58 -08:00
Kashif Rasul	fb33f35ffd	fix tests	2025-02-27 16:00:58 -08:00
Kashif Rasul	bce6405dea	remove flash_attention	2025-02-27 16:00:58 -08:00
Kashif Rasul	33cee018fd	formatting	2025-02-27 16:00:58 -08:00
Kashif Rasul	9739e4bf78	fixes for flash_attention	2025-02-27 16:00:58 -08:00
Kashif Rasul	4212ef81f3	add support for sdpa and flash_attention_2	2025-02-27 16:00:58 -08:00
Kashif Rasul	0d4325ee77	add weight init	2025-02-27 16:00:58 -08:00
Kashif Rasul	3bd0827e4d	fix isort	2025-02-27 16:00:58 -08:00
Kashif Rasul	b3198738a8	formatting	2025-02-27 16:00:58 -08:00
Kashif Rasul	8f6c2e1a7e	fix _quantile_loss	2025-02-27 16:00:58 -08:00
Kashif Rasul	c8703ff8b9	add loss	2025-02-27 16:00:58 -08:00
Kashif Rasul	f9e5db8df8	fix freq when window_size is given	2025-02-27 16:00:58 -08:00
Kashif Rasul	9f0f086a34	fix docstrings	2025-02-27 16:00:58 -08:00
Kashif Rasul	b437e87b88	rename script	2025-02-27 16:00:58 -08:00
Kashif Rasul	9e6750cc97	works with torch.Tensor inputs	2025-02-27 16:00:58 -08:00
Kashif Rasul	c3fbff2712	remove dropout_rate	2025-02-27 16:00:58 -08:00
Kashif Rasul	d7d3a1368b	add check_outputs	2025-02-27 16:00:58 -08:00
Kashif Rasul	c30e7488d2	initial script	2025-02-27 16:00:58 -08:00