Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,6 @@ from safetensors.torch import load_file
|
|
17 |
# local modules
|
18 |
from two_stream_shunt_adapter import TwoStreamShuntAdapter
|
19 |
from conditioning_shifter import ConditioningShifter, ShiftConfig, AdapterOutput
|
20 |
-
from embedding_manager import get_bank
|
21 |
from configs import T5_SHUNT_REPOS
|
22 |
|
23 |
warnings.filterwarnings("ignore")
|
@@ -26,7 +25,6 @@ warnings.filterwarnings("ignore")
|
|
26 |
dtype = torch.float16
|
27 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
28 |
|
29 |
-
_bank = get_bank() # singleton – optional caching
|
30 |
|
31 |
_t5_tok: Optional[T5Tokenizer] = None
|
32 |
_t5_mod: Optional[T5EncoderModel] = None
|
|
|
17 |
# local modules
|
18 |
from two_stream_shunt_adapter import TwoStreamShuntAdapter
|
19 |
from conditioning_shifter import ConditioningShifter, ShiftConfig, AdapterOutput
|
|
|
20 |
from configs import T5_SHUNT_REPOS
|
21 |
|
22 |
warnings.filterwarnings("ignore")
|
|
|
25 |
dtype = torch.float16
|
26 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
27 |
|
|
|
28 |
|
29 |
_t5_tok: Optional[T5Tokenizer] = None
|
30 |
_t5_mod: Optional[T5EncoderModel] = None
|