AbstractPhil commited on
Commit
d3593ef
Β·
1 Parent(s): 35e5214
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +2 -1
app.py CHANGED
@@ -6,8 +6,8 @@ from transformers import T5Tokenizer, T5EncoderModel
6
  from diffusers import DiffusionPipeline
7
  from safetensors.torch import load_file
8
  from huggingface_hub import hf_hub_download
9
- from two_stream_shunt_adapter import TwoStreamShuntAdapter
10
- from configs import T5_SHUNT_REPOS
11
 
12
  # ─── Device & Model Setup ─────────────────────────────────────
13
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
6
  from diffusers import DiffusionPipeline
7
  from safetensors.torch import load_file
8
  from huggingface_hub import hf_hub_download
9
+ from .app.two_stream_shunt_adapter import TwoStreamShuntAdapter
10
+ from .app.configs import T5_SHUNT_REPOS
11
 
12
  # ─── Device & Model Setup ─────────────────────────────────────
13
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
requirements.txt CHANGED
@@ -4,4 +4,5 @@ invisible_watermark
4
  torch
5
  transformers
6
  xformers
7
- matplotlib
 
 
4
  torch
5
  transformers
6
  xformers
7
+ matplotlib
8
+ sentencepiece