flux-labs

Running on Zero

vilarin commited on Jun 14

Commit

22d57bf

•

1 Parent(s): ee458f2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import gradio as gr
 import torch
 import numpy as np
 import random
-from diffusers import StableDiffusion3Pipeline, AutoencoderKL, SD3Transformer2DModel, StableDiffusion3Img2ImgPipeline
 import spaces
 from diffusers.utils import load_image
 from PIL import Image

 import torch
 import numpy as np
 import random
+from diffusers import StableDiffusion3Pipeline, AutoencoderKL, SD3Transformer2DModel, StableDiffusion3Img2ImgPipeline, FlowMatchEulerDiscreteScheduler
 import spaces
 from diffusers.utils import load_image
 from PIL import Image