search

QingruZhang / PASTA

PASTA: Post-hoc Attention Steering for LLMs
MIT License
108 stars 8 forks source link

Added support for GemmaForCausalLM #10

Closed Spongeorge closed 8 months ago

Spongeorge commented 8 months ago

For compatibility with Google's new Gemma models (https://ai.google.dev/gemma/docs)

QingruZhang commented 8 months ago

Thanks for your support on the development of pastalib!