Spaces:
Runtime error
Runtime error
init
Browse files- app.py +2 -2
- model/cnets.py +1 -3
- model/ea_model.py +0 -1
app.py
CHANGED
@@ -4,9 +4,9 @@ import time
|
|
4 |
import gradio as gr
|
5 |
import argparse
|
6 |
try:
|
7 |
-
from
|
8 |
except:
|
9 |
-
from
|
10 |
import torch
|
11 |
from fastchat.model import get_conversation_template
|
12 |
import re
|
|
|
4 |
import gradio as gr
|
5 |
import argparse
|
6 |
try:
|
7 |
+
from .model.ea_model import EaModel
|
8 |
except:
|
9 |
+
from model.ea_model import EaModel
|
10 |
import torch
|
11 |
from fastchat.model import get_conversation_template
|
12 |
import re
|
model/cnets.py
CHANGED
@@ -44,11 +44,9 @@ from transformers.utils import (
|
|
44 |
|
45 |
try:
|
46 |
from .configs import EConfig
|
47 |
-
from .choices import *
|
48 |
except:
|
49 |
from configs import EConfig
|
50 |
-
|
51 |
-
from utils import prepare_logits_processor
|
52 |
|
53 |
import time
|
54 |
|
|
|
44 |
|
45 |
try:
|
46 |
from .configs import EConfig
|
|
|
47 |
except:
|
48 |
from configs import EConfig
|
49 |
+
|
|
|
50 |
|
51 |
import time
|
52 |
|
model/ea_model.py
CHANGED
@@ -9,7 +9,6 @@ from .modeling_llama_kv import LlamaForCausalLM as KVLlamaForCausalLM
|
|
9 |
from .modeling_mixtral_kv import MixtralForCausalLM as KVMixtralForCausalLM
|
10 |
from .utils import *
|
11 |
from .kv_cache import initialize_past_key_values
|
12 |
-
from .choices import mc_sim_7b_63
|
13 |
from transformers import AutoTokenizer
|
14 |
import os
|
15 |
from huggingface_hub import hf_hub_download
|
|
|
9 |
from .modeling_mixtral_kv import MixtralForCausalLM as KVMixtralForCausalLM
|
10 |
from .utils import *
|
11 |
from .kv_cache import initialize_past_key_values
|
|
|
12 |
from transformers import AutoTokenizer
|
13 |
import os
|
14 |
from huggingface_hub import hf_hub_download
|