From 6a3edb05428c3f8121402b32dbe47f62065ddfd9 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Sat, 15 Jul 2023 22:40:55 -0700 Subject: [PATCH] Clean up llamacpp_hf.py --- modules/llamacpp_hf.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/modules/llamacpp_hf.py b/modules/llamacpp_hf.py index f031861f..12212fec 100644 --- a/modules/llamacpp_hf.py +++ b/modules/llamacpp_hf.py @@ -2,8 +2,6 @@ import os from pathlib import Path from typing import Any, Dict, Optional, Union -import llama_cpp -import numpy as np import torch from llama_cpp import Llama from torch.nn import CrossEntropyLoss @@ -11,7 +9,6 @@ from transformers import GenerationConfig, PretrainedConfig, PreTrainedModel from transformers.modeling_outputs import CausalLMOutputWithPast from modules import shared -from modules.llamacpp_model import LlamaCppModel from modules.logging_colors import logger