AbstractPhil commited on
Commit
c557c56
·
verified ·
1 Parent(s): 32a85ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
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