mirror of
https://github.com/PeiJieSun/DBDI-main.git
synced 2026-09-10 06:08:54 -07:00
PeijieSun
26 lines
879 B
Python
26 lines
879 B
Python
"""
|
|
Configuration file .
|
|
"""
|
|
|
|
# Data paths
|
|
path_harmful = "data/harmful.csv"
|
|
path_harmless = "data/harmless.csv"
|
|
path_harmful_test = "data/harmful_test.csv"
|
|
path_harmless_test = "data/harmless_test.csv"
|
|
path_harmful_calibration = "data/harmful_calibration.csv"
|
|
path_harmless_calibration = "data/harmless_calibration.csv"
|
|
|
|
# Model paths
|
|
model_paths = {
|
|
"mistral": "./models2/Mistral-7B-Instruct-v0.2",
|
|
"llama-2": "./models/Llama-2-7b-chat-hf",
|
|
"vicuna-7b": "./models/vicuna-7b-v1.5",
|
|
"gemma-2-9b-it": "./models2/gemma-2-9b-it",
|
|
"llama-3": "./models/Llama-3.1-8B-Instruct",
|
|
"llama-3.2-3b": "./models/Llama-3.2-3B-Instruct",
|
|
"Qwen7B": "./models/Qwen2.5-7B-Instruct",
|
|
# "mistral-sorry-bench": "./models/ft-mistral-7b-instruct-v0.2-sorry-bench-202406",
|
|
"deepseek":"./models2/deepseek-llm-7b-chat",
|
|
"Llamaguard":"./models2/Llama-Guard-3-8B"
|
|
}
|