LPX55 commited on
Commit
b3266d4
·
verified ·
1 Parent(s): a102a01

Update optimized.py

Browse files
Files changed (1) hide show
  1. optimized.py +2 -0
optimized.py CHANGED
@@ -2,10 +2,12 @@ import torch
2
  import spaces
3
  import os
4
  import diffusers
 
5
  from diffusers.utils import load_image
6
  from diffusers import FluxControlNetModel, FluxControlNetPipeline, AutoencoderKL
7
  import gradio as gr
8
  from accelerate import dispatch_model, infer_auto_device_map
 
9
  # Corrected and optimized FluxControlNet implementation
10
 
11
  def self_attention_slicing(module, slice_size=3):
 
2
  import spaces
3
  import os
4
  import diffusers
5
+ import PIL
6
  from diffusers.utils import load_image
7
  from diffusers import FluxControlNetModel, FluxControlNetPipeline, AutoencoderKL
8
  import gradio as gr
9
  from accelerate import dispatch_model, infer_auto_device_map
10
+ from PIL import Image
11
  # Corrected and optimized FluxControlNet implementation
12
 
13
  def self_attention_slicing(module, slice_size=3):