@@ -1,3 +1,6 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# This software may be used and distributed according to the terms of the Llama 2 Community License Agreement.
+
import sqlite3
import csv
import langchain
from langchain.llms import Replicate
import streamlit as st
import os
# Define the input and output file names
import argparse
import json
import logging
#!/bin/bash
# Prompt the user for the EVAL_PATH
import datasets
import re
import gc
import glob
import fire
from transformers import AutoTokenizer, AutoModelForCausalLM
import matplotlib.pyplot as plt
from finetuning_data_formatter import TrainingExample, Guidelines, Category, LlamaGuardPromptConfigs, LlamaGuardGenerationConfigs, ExplanationPosition, AugmentationConfigs, FormatterConfigs, create_formatted_finetuning_examples
# Define a list of guidelines and training examples to match
from typing import List, Union