barreloflube commited on
Commit
3a5022f
·
1 Parent(s): 030e485

Refactor flux_events.py and flux_helpers.py

Browse files
modules/events/flux_events.py CHANGED
@@ -8,7 +8,6 @@ from huggingface_hub import ModelCard
8
  from modules.helpers.flux_helpers import FluxReq, FluxImg2ImgReq, FluxInpaintReq, ControlNetReq, gen_img
9
  from config import flux_models, flux_loras
10
 
11
- flux_models = ["black-forest-labs/FLUX.1-dev"]
12
  loras = flux_loras
13
 
14
 
 
8
  from modules.helpers.flux_helpers import FluxReq, FluxImg2ImgReq, FluxInpaintReq, ControlNetReq, gen_img
9
  from config import flux_models, flux_loras
10
 
 
11
  loras = flux_loras
12
 
13
 
modules/helpers/flux_helpers.py CHANGED
@@ -16,7 +16,7 @@ from diffusers.schedulers import *
16
  from huggingface_hub import hf_hub_download
17
  from sd_embed.embedding_funcs import get_weighted_text_embeddings_flux1
18
 
19
- from .common_helpers import *
20
 
21
 
22
  def load_sd():
 
16
  from huggingface_hub import hf_hub_download
17
  from sd_embed.embedding_funcs import get_weighted_text_embeddings_flux1
18
 
19
+ from .common_helpers import ControlNetReq, BaseReq, BaseImg2ImgReq, BaseInpaintReq, cleanup, get_controlnet_images, resize_images
20
 
21
 
22
  def load_sd():