From a4495a8fd87e092980c7f3e129a84499c6f3fc9f Mon Sep 17 00:00:00 2001
From: Maximilian Schmidt <maximilian.schmidt@ims.uni-stuttgart.de>
Date: Fri, 17 May 2024 20:24:27 +0200
Subject: [PATCH] Remove PyTorch dependency

---
 task.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/task.py b/task.py
index d062253..abb26c3 100644
--- a/task.py
+++ b/task.py
@@ -5,15 +5,13 @@ from functools import lru_cache
 from statistics import mean
 from typing import Union
 
+from cli import argument_parser
 from datasets import Dataset, load_dataset
 from evaluate import load as load_metric
 from llama_cpp import LlamaGrammar, deque
-from torch.utils import data
-from tqdm import tqdm
-
-from cli import argument_parser
 from models import Llama2, LLMModel, OpenAI
 from opt_types import ModelUsage
+from tqdm import tqdm
 from utils import log_calls, logger
 
 SYSTEM_MESSAGE = """
-- 
GitLab