Jake Poznanski 07c0323c91 Adding lora config to try to address OOMs

2024-09-25 07:57:01 -07:00

..

config

Adding lora config to try to address OOMs

2024-09-25 07:57:01 -07:00

core

Loading dataset from config now

2024-09-23 09:40:24 -07:00

__init__.py

Starting to write dataloader for visual lm data

2024-09-18 21:42:09 +00:00

batch_inference.py

Enabling model eval

2024-09-24 10:48:53 -07:00

dataloader.py

typos

2024-09-23 09:43:36 -07:00

dataprep.py

More flexibility in dataloader dims

2024-09-24 19:47:13 -07:00

train.py

New images work much better now, and device map fix

2024-09-24 12:58:18 -07:00

utils.py

Importing core training config stuff from dolma refine

2024-09-19 21:55:07 +00:00