hysts HF staff commited on
Commit
8c7d74b
1 Parent(s): e77b7f5
Files changed (1) hide show
  1. app_sketch.py +2 -4
app_sketch.py CHANGED
@@ -14,8 +14,6 @@ from utils import (
14
  randomize_seed_fn,
15
  )
16
 
17
- SKETCH_ADAPTER_NAME = "TencentARC/t2i-adapter-sketch-sdxl-1.0"
18
-
19
 
20
  def create_demo(model: Model) -> gr.Blocks:
21
  def run(
@@ -40,7 +38,7 @@ def create_demo(model: Model) -> gr.Blocks:
40
  image=image,
41
  prompt=prompt,
42
  negative_prompt=negative_prompt,
43
- adapter_name=SKETCH_ADAPTER_NAME,
44
  num_inference_steps=num_steps,
45
  guidance_scale=guidance_scale,
46
  adapter_conditioning_scale=adapter_conditioning_scale,
@@ -160,6 +158,6 @@ def create_demo(model: Model) -> gr.Blocks:
160
 
161
 
162
  if __name__ == "__main__":
163
- model = Model(SKETCH_ADAPTER_NAME)
164
  demo = create_demo(model)
165
  demo.queue(max_size=20).launch()
 
14
  randomize_seed_fn,
15
  )
16
 
 
 
17
 
18
  def create_demo(model: Model) -> gr.Blocks:
19
  def run(
 
38
  image=image,
39
  prompt=prompt,
40
  negative_prompt=negative_prompt,
41
+ adapter_name="sketch",
42
  num_inference_steps=num_steps,
43
  guidance_scale=guidance_scale,
44
  adapter_conditioning_scale=adapter_conditioning_scale,
 
158
 
159
 
160
  if __name__ == "__main__":
161
+ model = Model("sketch")
162
  demo = create_demo(model)
163
  demo.queue(max_size=20).launch()