FEAT Add adversarial suffix attack GCG

.env_example

@@ -70,6 +70,8 @@ AZURE_ML_REGISTRY_NAME="azureml"
 # AZURE ML and HF Model Download/Register Compute Configuration
 # Update with your model ID
 HF_MODEL_ID="mistralai/Mistral-7B-Instruct-v0.1"
+HF_TOKEN="<Provide HuggingFace Token here>"
+
 # Update with your task name
 TASK_NAME="text-generation"
 AZURE_ML_COMPUTE_TYPE="amlcompute"

.gitignore

@@ -1,6 +1,7 @@
 # PyRIT-specific configs
 submodules/
 results/
+eval/
 default_memory.json.memory
 
 # Byte-compiled / optimized / DLL files

pyproject.toml

@@ -44,10 +44,12 @@ dependencies = [
  "confusables==1.2.0",
  "duckdb==0.10.0",
  "duckdb-engine==0.11.2",
+ "fschat==0.2.36",
  "jsonpickle>=3.0.4",
  "jupyter>=1.0.0",
  "ipykernel>=6.29.4",
  "logzero>=1.7.0",
+ "ml-collections==0.1.1",
  "numpy>=1.26.4",
  "onnxruntime>=1.14.1",
  "onnx>=1.16.0",
@@ -80,6 +82,9 @@ dev = [
  "semantic-kernel==0.9.4b1",
  "types-PyYAML>=6.0.12.9",
 ]
+torch = [
+ "torch>=2.3.0"
+]
 
 [tool.pytest.ini_options]
 pythonpath = ["."]