Update app.py
Browse files
app.py
CHANGED
@@ -1,16 +1,10 @@
|
|
1 |
import gradio as gr
|
2 |
from optimum.intel import OVDiffusionPipeline
|
3 |
-
from transformers import AutoTokenizer
|
4 |
from threading import Lock
|
5 |
|
6 |
-
# Load the pipeline
|
7 |
model_id = "OpenVINO/FLUX.1-schnell-int4-ov"
|
8 |
pipeline = OVDiffusionPipeline.from_pretrained(model_id, device="CPU")
|
9 |
-
|
10 |
-
# Explicitly load the tokenizer with fast settings
|
11 |
-
tokenizer = AutoTokenizer.from_pretrained("t5-v1_1-xxl", use_fast=True, add_prefix_space=True)
|
12 |
-
pipeline.text_encoder_2.tokenizer = tokenizer # Assign the tokenizer to the T5 encoder
|
13 |
-
|
14 |
lock = Lock()
|
15 |
|
16 |
# Define the image generation function
|
@@ -32,4 +26,4 @@ interface = gr.Interface(
|
|
32 |
|
33 |
# Launch the interface
|
34 |
if __name__ == "__main__":
|
35 |
-
interface.launch()
|
|
|
1 |
import gradio as gr
|
2 |
from optimum.intel import OVDiffusionPipeline
|
|
|
3 |
from threading import Lock
|
4 |
|
5 |
+
# Load the pipeline globally (includes the correct tokenizer)
|
6 |
model_id = "OpenVINO/FLUX.1-schnell-int4-ov"
|
7 |
pipeline = OVDiffusionPipeline.from_pretrained(model_id, device="CPU")
|
|
|
|
|
|
|
|
|
|
|
8 |
lock = Lock()
|
9 |
|
10 |
# Define the image generation function
|
|
|
26 |
|
27 |
# Launch the interface
|
28 |
if __name__ == "__main__":
|
29 |
+
interface.launch()
|