vikarti.anatra/blt

mirror of https://github.com/facebookresearch/blt.git synced 2025-01-18 16:37:46 +00:00

Author	SHA1	Message	Date
Pedro Rodriguez	7f305b3871	[WIP] Changes for training entropy model and correcting attention in local models Some checks are pending Lint with Black / lint (push) Waiting to run Details Lint with isort / lint (push) Waiting to run Details Summary: - Refactor local model configs to be separate and clearer - Add attention arguments and correct which attention is used in local models - Preparation for being able to have an entropy train script - Fix failing unit tests Test Plan:	2025-01-17 22:21:51 +00:00
Ink	caec8d2621	allow flex-attention to be disabled (#19 ) Some checks failed Lint with Black / lint (push) Has been cancelled Details Lint with isort / lint (push) Has been cancelled Details * allow flex-attention to silently fail * allow flex-attn to be disabled via an env var	2025-01-14 09:32:07 -08:00
Pedro Rodriguez	1da3dd9315	Update preprocess_entropies script to blt inference + add fsspec support (#23 ) Some checks are pending Lint with Black / lint (push) Waiting to run Details Lint with isort / lint (push) Waiting to run Details Summary: Test Plan:	2025-01-13 15:28:14 -08:00
Pedro Rodriguez	b0120da72f	Replace regular filesystem calls with fsspec + add s3 support (#18 ) Some checks failed Lint with Black / lint (push) Has been cancelled Details Lint with isort / lint (push) Has been cancelled Details Summary: For compatibility with either local/nfs or S3 datasets, swap to fsspec. Add a tool to compare local and remote filesystems Test Plan: - Ran regular train script - Ran with config with data in S3	2025-01-10 11:04:41 -08:00
Pedro Rodriguez	d4ddb95322	Add plotting code from paper (#17 ) Some checks are pending Lint with Black / lint (push) Waiting to run Details Lint with isort / lint (push) Waiting to run Details Summary: Test Plan:	2025-01-09 12:11:50 -08:00
Pedro Rodriguez	bcc039bb75	Initial commit	2024-12-12 15:32:30 -08:00

6 commits