├── .gitignore ├── AI_ETHICS.md ├── CODEOWNERS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE.txt ├── README.md ├── SECURITY.md ├── assets ├── complexity.png ├── intro.png ├── longbench.png └── needle.png ├── eval ├── LongBench │ ├── README.md │ ├── config │ │ ├── dataset2maxlen.json │ │ ├── dataset2prompt.json │ │ ├── model2maxlen.json │ │ └── model2path.json │ ├── eval.py │ ├── metrics.py │ └── pred.py └── needle │ ├── PaulGrahamEssays │ ├── addiction.txt │ ├── aord.txt │ ├── apple.txt │ ├── avg.txt │ ├── before.txt │ ├── bias.txt │ ├── boss.txt │ ├── copy.txt │ ├── corpdev.txt │ ├── desres.txt │ ├── diff.txt │ ├── ecw.txt │ ├── founders.txt │ ├── foundervisa.txt │ ├── gap.txt │ ├── gba.txt │ ├── gh.txt │ ├── goodtaste.txt │ ├── hubs.txt │ ├── iflisp.txt │ ├── island.txt │ ├── know.txt │ ├── langdes.txt │ ├── laundry.txt │ ├── love.txt │ ├── mod.txt │ ├── newideas.txt │ ├── nft.txt │ ├── philosophy.txt │ ├── popular.txt │ ├── pow.txt │ ├── rootsoflisp.txt │ ├── rss.txt │ ├── siliconvalley.txt │ ├── startuplessons.txt │ ├── submarine.txt │ ├── sun.txt │ ├── superangels.txt │ ├── todo.txt │ ├── unions.txt │ ├── useful.txt │ ├── vb.txt │ ├── vcsqueeze.txt │ ├── vw.txt │ ├── want.txt │ ├── web20.txt │ ├── weird.txt │ ├── wisdom.txt │ └── worked.txt │ ├── README.md │ ├── needle_in_haystack.py │ ├── utils.py │ └── visualize.py ├── my_baseline ├── GemFilter │ ├── gem_filter_utils.py │ ├── llama_select_attention.py │ ├── mistral_select_attention.py │ └── phi3_select_attention.py ├── H2O │ ├── llama_h2o_attention.py │ ├── mistral_h2o_attention.py │ ├── phi3_h2o_attention.py │ └── utils_h2o.py └── SnapKV │ ├── llama_hijack_4_43.py │ ├── mistral_hijack_4_43.py │ ├── monkeypatch.py │ ├── phi3_hijack_4_43.py │ └── snapkv_utils.py ├── my_utils ├── load_model.py └── my_generation.py ├── needle_eval.py ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- 1 | .vscode 2 | 3 | **/*.pyc 4 | -------------------------------------------------------------------------------- /AI_ETHICS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/AI_ETHICS.md -------------------------------------------------------------------------------- /CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/CODEOWNERS -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/SECURITY.md -------------------------------------------------------------------------------- /assets/complexity.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/assets/complexity.png -------------------------------------------------------------------------------- /assets/intro.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/assets/intro.png -------------------------------------------------------------------------------- /assets/longbench.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/assets/longbench.png -------------------------------------------------------------------------------- /assets/needle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/assets/needle.png -------------------------------------------------------------------------------- /eval/LongBench/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/LongBench/README.md -------------------------------------------------------------------------------- /eval/LongBench/config/dataset2maxlen.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/LongBench/config/dataset2maxlen.json -------------------------------------------------------------------------------- /eval/LongBench/config/dataset2prompt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/LongBench/config/dataset2prompt.json -------------------------------------------------------------------------------- /eval/LongBench/config/model2maxlen.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/LongBench/config/model2maxlen.json -------------------------------------------------------------------------------- /eval/LongBench/config/model2path.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/LongBench/config/model2path.json -------------------------------------------------------------------------------- /eval/LongBench/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/LongBench/eval.py -------------------------------------------------------------------------------- /eval/LongBench/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/LongBench/metrics.py -------------------------------------------------------------------------------- /eval/LongBench/pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/LongBench/pred.py -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/addiction.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/addiction.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/aord.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/aord.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/apple.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/apple.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/avg.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/avg.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/before.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/before.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/bias.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/bias.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/boss.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/boss.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/copy.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/copy.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/corpdev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/corpdev.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/desres.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/desres.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/diff.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/diff.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/ecw.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/ecw.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/founders.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/founders.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/foundervisa.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/foundervisa.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/gap.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/gap.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/gba.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/gba.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/gh.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/gh.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/goodtaste.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/goodtaste.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/hubs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/hubs.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/iflisp.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/iflisp.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/island.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/island.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/know.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/know.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/langdes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/langdes.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/laundry.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/laundry.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/love.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/love.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/mod.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/mod.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/newideas.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/newideas.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/nft.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/nft.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/philosophy.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/philosophy.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/popular.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/popular.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/pow.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/pow.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/rootsoflisp.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/rootsoflisp.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/rss.txt: -------------------------------------------------------------------------------- 1 | Aaron Swartz created a scraped 2 | feed 3 | of the essays page. -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/siliconvalley.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/siliconvalley.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/startuplessons.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/startuplessons.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/submarine.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/submarine.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/sun.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/sun.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/superangels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/superangels.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/todo.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/todo.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/unions.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/unions.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/useful.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/useful.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/vb.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/vb.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/vcsqueeze.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/vcsqueeze.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/vw.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/vw.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/want.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/want.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/web20.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/web20.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/weird.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/weird.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/wisdom.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/wisdom.txt -------------------------------------------------------------------------------- /eval/needle/PaulGrahamEssays/worked.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/PaulGrahamEssays/worked.txt -------------------------------------------------------------------------------- /eval/needle/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/README.md -------------------------------------------------------------------------------- /eval/needle/needle_in_haystack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/needle_in_haystack.py -------------------------------------------------------------------------------- /eval/needle/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/utils.py -------------------------------------------------------------------------------- /eval/needle/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/eval/needle/visualize.py -------------------------------------------------------------------------------- /my_baseline/GemFilter/gem_filter_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/GemFilter/gem_filter_utils.py -------------------------------------------------------------------------------- /my_baseline/GemFilter/llama_select_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/GemFilter/llama_select_attention.py -------------------------------------------------------------------------------- /my_baseline/GemFilter/mistral_select_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/GemFilter/mistral_select_attention.py -------------------------------------------------------------------------------- /my_baseline/GemFilter/phi3_select_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/GemFilter/phi3_select_attention.py -------------------------------------------------------------------------------- /my_baseline/H2O/llama_h2o_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/H2O/llama_h2o_attention.py -------------------------------------------------------------------------------- /my_baseline/H2O/mistral_h2o_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/H2O/mistral_h2o_attention.py -------------------------------------------------------------------------------- /my_baseline/H2O/phi3_h2o_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/H2O/phi3_h2o_attention.py -------------------------------------------------------------------------------- /my_baseline/H2O/utils_h2o.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/H2O/utils_h2o.py -------------------------------------------------------------------------------- /my_baseline/SnapKV/llama_hijack_4_43.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/SnapKV/llama_hijack_4_43.py -------------------------------------------------------------------------------- /my_baseline/SnapKV/mistral_hijack_4_43.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/SnapKV/mistral_hijack_4_43.py -------------------------------------------------------------------------------- /my_baseline/SnapKV/monkeypatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/SnapKV/monkeypatch.py -------------------------------------------------------------------------------- /my_baseline/SnapKV/phi3_hijack_4_43.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/SnapKV/phi3_hijack_4_43.py -------------------------------------------------------------------------------- /my_baseline/SnapKV/snapkv_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_baseline/SnapKV/snapkv_utils.py -------------------------------------------------------------------------------- /my_utils/load_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_utils/load_model.py -------------------------------------------------------------------------------- /my_utils/my_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/my_utils/my_generation.py -------------------------------------------------------------------------------- /needle_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/needle_eval.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/GemFilter/HEAD/setup.py --------------------------------------------------------------------------------