Prompting is still in progress, but a more complete pipeline has been developed with functions in modules/prompt_fns.py and modules/pipeline_fns.py. These functions are applied in the script src/gen_pipeline.py.
Models llama2-7b, falcon-7b, T5 and StableBeluga-7b have been tested.
CLEANING
Stories has been cleaned further as tokens (e.g., [ wp ]) interfered with some model completions. Dailydialog likely needs similar cleaning in the future (removing [EOT] tokens).
FURTHER DEVELOPMENT
Task prompting (e.g., "summarise this") may still need tweaks
Further implementation is needed to get the chat versions of llama to run (wrapping task prompts in a chat interface)
Progress Update
PROMPTING
Prompting is still in progress, but a more complete pipeline has been developed with functions in
modules/prompt_fns.py
andmodules/pipeline_fns.py
. These functions are applied in the scriptsrc/gen_pipeline.py
.Models
llama2-7b
,falcon-7b
,T5
andStableBeluga-7b
have been tested.CLEANING
Stories
has been cleaned further as tokens (e.g.,[ wp ]
) interfered with some model completions. Dailydialog likely needs similar cleaning in the future (removing[EOT]
tokens).FURTHER DEVELOPMENT
chat
versions ofllama
to run (wrapping task prompts in a chat interface)