Spaces:
Runtime error
Runtime error
lionelgarnier
commited on
Commit
·
57f3ad5
1
Parent(s):
33188dc
Switch to fast tokenizer and reorder prompt refinement inputs
Browse files
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=
|
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'
|