init

app.py

@@ -4,9 +4,9 @@ import time
 import gradio as gr
 import argparse
 try:
-    from ..model.ea_model import EaModel
+    from .model.ea_model import EaModel
 except:
-    from eagle.model.ea_model import EaModel
+    from model.ea_model import EaModel
 import torch
 from fastchat.model import get_conversation_template
 import re

model/cnets.py

@@ -44,11 +44,9 @@ from transformers.utils import (
 
 try:
     from .configs import EConfig
-    from .choices import *
 except:
     from configs import EConfig
-    from choices import *
-    from utils import prepare_logits_processor
+
 
 import time
 

model/ea_model.py

@@ -9,7 +9,6 @@ from .modeling_llama_kv import LlamaForCausalLM as KVLlamaForCausalLM
 from .modeling_mixtral_kv import MixtralForCausalLM as KVMixtralForCausalLM
 from .utils import *
 from .kv_cache import initialize_past_key_values
-from .choices import mc_sim_7b_63
 from transformers import AutoTokenizer
 import os
 from huggingface_hub import hf_hub_download