Update app.py

app.py

@@ -6,9 +6,16 @@ import json
 import torch
 import spaces
 
-from diffusers import AutoencoderKL, SD3Transformer2DModel, StableDiffusion3Pipeline
-from diffusers.loaders.single_file_utils import convert_sd3_transformer_checkpoint_to_diffusers
 from huggingface_hub import hf_hub_download
+from diffusers import (
+    AutoencoderKL,
+    SD3Transformer2DModel,
+    StableDiffusion3Pipeline,
+    FlowMatchEulerDiscreteScheduler
+)
+from diffusers.loaders.single_file_utils import (
+    convert_sd3_transformer_checkpoint_to_diffusers,
+)
 from transformers import (
     CLIPTextModelWithProjection,
     CLIPTokenizer,
@@ -57,7 +64,7 @@ text_encoder_3 = T5EncoderModel.from_pretrained(model_repo_id, subfolder="text_e
 tokenizer = CLIPTokenizer.from_pretrained(model_repo_id, subfolder="tokenizer")
 tokenizer_2 = CLIPTokenizer.from_pretrained(model_repo_id, subfolder="tokenizer_2")
 tokenizer_3 = T5Tokenizer.from_pretrained(model_repo_id, subfolder="tokenizer_3")
-
+scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo_id, subfolder="scheduler")
 
 # Create pipeline from our models
 pipe = StableDiffusion3Pipeline(