Add chat templates to README

#14
by Rocketknight1 HF staff - opened
No description provided.
Google org

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

Sign up or log in to comment