flux-labs

Running on Zero

vilarin commited on Jun 13

Commit

4b9fa98

•

1 Parent(s): 5d300f8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import gradio as gr
 import torch
 import numpy as np
 import random
-from diffusers import StableDiffusion3Pipeline, SD3Transformer2DModel, FlowMatchEulerDiscreteScheduler
 import spaces
 from PIL import Image
 import requests

 import torch
 import numpy as np
 import random
+from diffusers import StableDiffusion3Pipeline, AutoencoderKL, SD3Transformer2DModel, FlowMatchEulerDiscreteScheduler
 import spaces
 from PIL import Image
 import requests