multimodalart HF Staff commited on
Commit
f50f0c3
·
verified ·
1 Parent(s): 3c0d4f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,11 +1,10 @@
1
  import gradio as gr
2
  import torch
3
  import spaces
4
- from diffusers import AutoencoderKL, FluxTransformer2DModel
5
  from diffusers.utils import load_image
6
  from controlnet_flux import FluxControlNetModel
7
  from transformer_flux import FluxTransformer2DModel
8
- from pipeline_flux_controlnet_inpaint import FluxControlNetInpaintingPipeline
9
  from transformers import T5EncoderModel, CLIPTextModel
10
  from PIL import Image, ImageDraw
11
  import numpy as np
@@ -18,7 +17,7 @@ transformer = FluxTransformer2DModel.from_pretrained(
18
  "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16
19
  )
20
 
21
- pipe = FluxControlNetInpaintingPipeline.from_pretrained(
22
  "black-forest-labs/FLUX.1-dev",
23
  transformer=transformer,
24
  controlnet=controlnet,
 
1
  import gradio as gr
2
  import torch
3
  import spaces
4
+ from diffusers import AutoencoderKL, FluxTransformer2DModel, FluxControlNetInpaintPipeline
5
  from diffusers.utils import load_image
6
  from controlnet_flux import FluxControlNetModel
7
  from transformer_flux import FluxTransformer2DModel
 
8
  from transformers import T5EncoderModel, CLIPTextModel
9
  from PIL import Image, ImageDraw
10
  import numpy as np
 
17
  "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16
18
  )
19
 
20
+ pipe = FluxControlNetInpaintPipeline.from_pretrained(
21
  "black-forest-labs/FLUX.1-dev",
22
  transformer=transformer,
23
  controlnet=controlnet,