Update eval_epochs.py

llm_toolkit/eval_epochs.py

@@ -122,8 +122,7 @@ if __name__ == "__main__":
         print(f"(0) GPU = {gpu_stats.name}. Max memory = {max_memory} GB.")
         print(f"{start_gpu_memory} GB of memory reserved.")
 
-    # model, tokenizer = load_model(model_name, load_in_4bit=load_in_4bit)
-    model, tokenizer = None, None
+    model, tokenizer = load_model(model_name, load_in_4bit=load_in_4bit)
 
     datasets = load_translation_dataset(data_path, tokenizer, num_shots=0)
     print_row_details(datasets["test"].to_pandas())