Update mini.py

mini.py

@@ -5,8 +5,8 @@ from PIL import Image
 import os
 from transformers import CLIPTokenizer, CLIPTextModel, AutoProcessor, T5EncoderModel, T5TokenizerFast
 from diffusers import AutoencoderKL, FlowMatchEulerDiscreteScheduler
-from flux.transformer_flux import FluxTransformer2DModel
-from flux.pipeline_flux_chameleon import FluxPipeline
+from flux.transformer_flux_simple import FluxTransformer2DModel
+from flux.pipeline_flux_chameleon_og import FluxPipeline
 import torch.nn as nn
 import math
 import logging