fix paths

modules/llm/Mistral-7b/Mistral-7b_zero_shot.py

@@ -5,11 +5,12 @@ from alive_progress import alive_bar
 import torch
 import transformers
 import bitsandbytes, flash_attn
+
 spec = importlib.util.spec_from_file_location("utils", "../utils.py")
 utils = importlib.util.module_from_spec(spec)
 spec.loader.exec_module(utils)
 
-model_id = "../../models/Mistral-7B-Instruct-v0.2"
+model_id = "../../../models/Mistral-7B-Instruct-v0.2"
 
 model = transformers.AutoModelForCausalLM.from_pretrained(
     model_id,

modules/llm/Mixtral-8x7b/Mixtral-8x7b_zero_shot.py

@@ -9,7 +9,7 @@ spec = importlib.util.spec_from_file_location("utils", "../utils.py")
 utils = importlib.util.module_from_spec(spec)
 spec.loader.exec_module(utils)
 
-model_id = "../../models/Mixtral-8x7B-Instruct-v0.1"
+model_id = "../../../models/Mixtral-8x7B-Instruct-v0.1"
 
 model = transformers.AutoModelForCausalLM.from_pretrained(
     model_id,