11 Commits

This Branch

This Branch

All Branches

Author	SHA1	Message	Date
Kashif Rasul	d2cb4849b6	fix docs string	2026-03-11 17:38:09 +01:00
Kashif Rasul	0134e55a6e	Replace ModelHubMixin with PytorchModelHubMixin	2026-03-11 09:36:14 +01:00
Kashif Rasul	a8bcd5d6d9	use the ModelHubMixin api	2026-03-06 10:53:18 +01:00
Kashif Rasul	177ab03e9c	fix v1 jax version	2026-02-18 14:50:56 +01:00
Kashif Rasul	705685c912	fix v1 var calculation ... 1. Masked variance calculation (lines 95-107): Changed from the numerically unstable E[X²] - E[X]² formula to the stable centered formula E[(X-μ)²] 2. Sigma clamping (line 609): Changed from torch.where(sigma < tolerance, 1.0, sigma) to torch.clamp(sigma, min=tolerance)	2025-11-25 09:40:07 +01:00
Kashif Rasul	2a83a0f82d	add back checkpoint_path	2024-05-10 12:23:25 +02:00
Kashif Rasul	4a2b0ae837	fix checkpoint_path	2024-05-10 12:01:42 +02:00
Kashif Rasul	f04d4d637d	update readme	2024-05-10 11:39:38 +02:00
Kashif Rasul	d5f55b0825	use snapshot_download	2024-05-10 11:32:53 +02:00
Kashif Rasul	afeb84f77a	use repo_id to download checkpoints	2024-05-10 10:48:49 +02:00
Kashif Rasul	6e38ec85bb	typos	2024-05-10 09:25:17 +02:00