Yeu-Tong Lau

[Bug Report] Gemma-2-2b-it output logit doesn't match with huggingface

3

**Describe the bug** The output logits from transformer_lens and huggingface are quite different using Gemma-2-2b-it model **Code example** ```python import torch import transformer_lens from transformers import AutoTokenizer, AutoModelForCausalLM device =...