Remove PyTorch dependency

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 = """