ironjr commited on
Commit
af7ced6
·
verified ·
1 Parent(s): ce3f8a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,7 +40,7 @@ import gradio as gr
40
  from huggingface_hub import snapshot_download
41
  import spaces
42
 
43
- from model import StableMultiDiffusionSDXLPipeline
44
  from util import seed_everything
45
  from prompt_util import preprocess_prompts, _quality_dict, _style_dict
46
  from share_btn import share_js
@@ -108,7 +108,7 @@ else:
108
  model_dict = {os.path.splitext(os.path.basename(opt.model))[0]: opt.model}
109
 
110
  models = {
111
- k: StableMultiDiffusionSDXLPipeline(device, hf_key=v, has_i2t=False).cuda()
112
  for k, v in model_dict.items()
113
  }
114
 
 
40
  from huggingface_hub import snapshot_download
41
  import spaces
42
 
43
+ from model import SemanticDrawSDXLPipeline
44
  from util import seed_everything
45
  from prompt_util import preprocess_prompts, _quality_dict, _style_dict
46
  from share_btn import share_js
 
108
  model_dict = {os.path.splitext(os.path.basename(opt.model))[0]: opt.model}
109
 
110
  models = {
111
+ k: SemanticDrawSDXLPipeline(device, hf_key=v, has_i2t=False).cuda()
112
  for k, v in model_dict.items()
113
  }
114