Mohit Sharma f9e561eced Update ROCM libs and improvements (#2579) ... * style * update torch * ix issues * fix clone * revert mkl * added custom PA * style * fix style * style * hide env vart * fix mixtral model * add skinny kernel and merge fixes * fixed style * fix issue for sliding window models * addressed review comments * fix import * improved error messag * updated default value * remove import * fix imports after rebase * float16 dep * improve dockerfile * cleaned dockerfile		2024-09-30 10:54:32 +02:00
..
custom_modeling	Update ROCM libs and improvements (#2579)	2024-09-30 10:54:32 +02:00
__init__.py	Add support for scalar FP8 weight scales (#2550)	2024-09-24 13:57:40 +02:00
bloom.py	Refactor dead code - Removing all `flash_xxx.py` files. (#2166)	2024-07-05 10:29:56 +02:00
causal_lm.py	Fixing exl2 and other quanize tests again. (#2419)	2024-08-15 11:12:51 +02:00
flash_causal_lm.py	Update ROCM libs and improvements (#2579)	2024-09-30 10:54:32 +02:00
galactica.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
globals.py	Lots of improvements (Still 2 allocators) (#2449)	2024-08-29 16:29:01 +02:00
idefics.py	Upgrading exl2. (#2415)	2024-08-14 11:58:08 +02:00
idefics_causal_lm.py	Upgrading exl2. (#2415)	2024-08-14 11:58:08 +02:00
mamba.py	Fixing exl2 and other quanize tests again. (#2419)	2024-08-15 11:12:51 +02:00
model.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
pali_gemma.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
seq2seq_lm.py	Fixing exl2 and other quanize tests again. (#2419)	2024-08-15 11:12:51 +02:00
types.py	feat: add ruff and resolve issue (#2262)	2024-07-26 10:29:09 -04:00
vlm_causal_lm.py	Prefix test - Different kind of load test to trigger prefix test bugs. (#2490)	2024-09-11 18:10:40 +02:00