From e1f1c2100e0ddd52469748d1d825986ca2750f29 Mon Sep 17 00:00:00 2001 From: marcusm117 Date: Sun, 5 May 2024 13:26:04 -0400 Subject: [PATCH] udpate run_identity_chain_huggingface.py --- examples/run_identity_chain_huggingface.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/examples/run_identity_chain_huggingface.py b/examples/run_identity_chain_huggingface.py index 4e3f37e..b6be297 100644 --- a/examples/run_identity_chain_huggingface.py +++ b/examples/run_identity_chain_huggingface.py @@ -11,6 +11,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, set_seed # Internal Modules +# INSTRUCTION_MODELS and FOUNDATION_MODELS record the model names supported by +# run_identity_chain_huggingface.py. If you want to adapt this script to other +# huggingface or local models, add the names to the corresponding list. from identitychain import IdentityChain, INSTRUCTION_MODELS, FOUNDATION_MODELS from identitychain.dialogue import ( DialogueTemplate,