14 Commits

Include renames

Author	SHA1	Message	Date
Jaret Burkett	e20a869dc1	Update docker install for blacwell	2025-06-01 19:06:53 -06:00
Jaret Burkett	12fa109910	Updated cuda arch list on docker build	2025-06-01 13:37:11 -06:00
Jaret Burkett	64130189ce	Bumped torch and cuda to support blackwell arch	2025-05-09 11:17:58 -06:00
Jaret Burkett	1628884254	Remove submodule install from docker	2025-04-18 10:41:52 -06:00
Jaret Burkett	524bd2edfc	Make flash attn optional. Handle larger batch sizes.	2025-04-14 14:34:46 +00:00
Jaret Burkett	f80cf99f40	Hidream is training, but has a memory leak	2025-04-13 23:28:18 +00:00
Jaret Burkett	3c95f87a90	Added some missing dependencies	2025-03-27 18:48:48 -06:00
Jaret Burkett	1d5f387f54	Fix docker command to work better with runpod	2025-03-27 17:44:46 -06:00
Jaret Burkett	71d7a52146	Fixed issue with python being wrong on docker	2025-03-24 14:16:09 -06:00
Jaret Burkett	6c5eb0cf87	Move pytorch install above cache bust to prevent reinstalling and reuploading it	2025-03-21 06:43:09 -06:00
Jaret Burkett	aceb3a0f25	Rework docker	2025-03-20 20:07:27 -06:00
Jaret Burkett	f85ad452c6	Added initial support for pixtral vision as a vision encoder.	2024-09-21 15:21:14 -06:00
Jaret Burkett	dd889086f4	Updates to the docker file for jupyterlab	2024-09-21 12:07:07 -06:00
Jaret Burkett	88acc28d7f	Prep for runpod docker	2024-08-03 12:41:06 -06:00