mrq 70cf694cfd output attention scores for SDPA/flash, since naive attention seems broken		2024-10-12 12:09:17 -05:00
..
emb
engines
ext
models	output attention scores for SDPA/flash, since naive attention seems broken	2024-10-12 12:09:17 -05:00
utils
__init__.py
__main__.py	modified demo page to be more modular with demoing comparisons, actually provide a path to use modified naive attention, entropix sampling is not tied to an experimental yaml flag now	2024-10-12 11:27:55 -05:00
config.py	modified demo page to be more modular with demoing comparisons, actually provide a path to use modified naive attention, entropix sampling is not tied to an experimental yaml flag now	2024-10-12 11:27:55 -05:00
data.py
demo.py	output attention scores for SDPA/flash, since naive attention seems broken	2024-10-12 12:09:17 -05:00
export.py
inference.py	modified demo page to be more modular with demoing comparisons, actually provide a path to use modified naive attention, entropix sampling is not tied to an experimental yaml flag now	2024-10-12 11:27:55 -05:00
plot.py
samplers.py	modified demo page to be more modular with demoing comparisons, actually provide a path to use modified naive attention, entropix sampling is not tied to an experimental yaml flag now	2024-10-12 11:27:55 -05:00
train.py
webui.py	modified demo page to be more modular with demoing comparisons, actually provide a path to use modified naive attention, entropix sampling is not tied to an experimental yaml flag now	2024-10-12 11:27:55 -05:00