Update main.py

main.py

@@ -26,16 +26,16 @@ MODEL_CONFIGS = {
         "default_concept": "humor",
         "concepts": ["humor", "creativity", "quality", "truthfulness", "compliance"],
     },
-    # "Mistral-7B-Instruct-v0.1": {
-    #     "identifier": "mistralai/Mistral-7B-Instruct-v0.1",
-    #     "dtype": torch.bfloat16 if device.type == "cuda" else torch.float32,
-    #     "guidance_interval": [-128.0, 128.0],
-    #     "default_guidance_scale": 48.0,
-    #     "min_guidance_layer": 8,
-    #     "max_guidance_layer": 32,
-    #     "default_concept": "humor",
-    #     "concepts": ["humor", "creativity", "quality", "truthfulness", "compliance"],
-    # },
+    "Mistral-7B-Instruct-v0.1": {
+        "identifier": "mistralai/Mistral-7B-Instruct-v0.1",
+        "dtype": torch.bfloat16 if device.type == "cuda" else torch.float32,
+        "guidance_interval": [-128.0, 128.0],
+        "default_guidance_scale": 48.0,
+        "min_guidance_layer": 8,
+        "max_guidance_layer": 32,
+        "default_concept": "humor",
+        "concepts": ["humor", "creativity", "quality", "truthfulness", "compliance"],
+    },
 }
 
 def load_concept_vectors(model, concepts):