lionelgarnier commited on
Commit
57f3ad5
·
1 Parent(s): 33188dc

Switch to fast tokenizer and reorder prompt refinement inputs

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,7 +27,7 @@ def get_image_gen_pipeline():
27
  # Load fast tokenizer for the image pipeline
28
  tokenizer = AutoTokenizer.from_pretrained(
29
  "black-forest-labs/FLUX.1-schnell",
30
- use_fast=False
31
  )
32
  _image_gen_pipeline = DiffusionPipeline.from_pretrained(
33
  "black-forest-labs/FLUX.1-schnell",
@@ -306,8 +306,8 @@ def create_interface():
306
  "medieval flip flops",
307
  "cat shaped cake mold"
308
  ],
309
- inputs = [prompt],
310
  fn=refine_prompt,
 
311
  outputs = [refined_prompt],
312
  cache_examples=True,
313
  cache_mode='lazy'
 
27
  # Load fast tokenizer for the image pipeline
28
  tokenizer = AutoTokenizer.from_pretrained(
29
  "black-forest-labs/FLUX.1-schnell",
30
+ use_fast=True
31
  )
32
  _image_gen_pipeline = DiffusionPipeline.from_pretrained(
33
  "black-forest-labs/FLUX.1-schnell",
 
306
  "medieval flip flops",
307
  "cat shaped cake mold"
308
  ],
 
309
  fn=refine_prompt,
310
+ inputs = [prompt],
311
  outputs = [refined_prompt],
312
  cache_examples=True,
313
  cache_mode='lazy'