Spaces:
Running
on
Zero
Running
on
Zero
Update optimized.py
Browse files- 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):
|