Text Generation
Transformers
Safetensors
gemma2
conversational
text-generation-inference

Add chat templates to README

#14
by Rocketknight1 HF Staff - opened
No description provided.

Per https://huggingface.co/google/shieldgemma-2b/discussions/9, we need to add from torch.nn.functional import softmax to this example to make it copy-paste-able.

Otherwise, LGTM.

@RyanMullins it should be copy-pastable already, because it uses torch.softmax and torch is imported!

RyanMullins changed pull request status to merged
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment