Update app.py

app.py

@@ -45,7 +45,8 @@ from diffusers import StableDiffusionPipeline
 from diffusers import DiffusionPipeline
 from diffusers import EulerDiscreteScheduler
 from diffusers import UNet2DConditionModel
-diffusers.FlowMatchEulerDiscreteScheduler = EulerDiscreteScheduler
+from diffusers import FlowMatchEulerDiscreteScheduler
+# diffusers.FlowMatchEulerDiscreteScheduler = EulerDiscreteScheduler
 
 import transformers.utils.import_utils as _import_utils
 from transformers.utils import is_flash_attn_2_available
@@ -80,10 +81,10 @@ scheduler = EulerDiscreteScheduler.from_pretrained(
     torch_dtype=torch.float16,
 )
 text_encoder      = CLIPTextModel.from_pretrained(
-    model_repo, subfolder="text_encoder_1", torch_dtype=torch.float16
+    model_repo, subfolder="text_encoder", torch_dtype=torch.float16
 )
 tokenizer         = CLIPTokenizer.from_pretrained(
-    model_repo, subfolder="tokenizer_1"
+    model_repo, subfolder="tokenizer"
 )
 feature_extractor = CLIPFeatureExtractor.from_pretrained(
     model_repo, subfolder="feature_extractor"