JiantaoLin
commited on
Commit
Β·
c6de8f2
1
Parent(s):
18b35a5
new
Browse files
pipeline/kiss3d_wrapper.py
CHANGED
|
@@ -29,7 +29,7 @@ from transformers import AutoProcessor, AutoModelForCausalLM, AutoTokenizer
|
|
| 29 |
from models.llm.llm import load_llm_model, get_llm_response
|
| 30 |
|
| 31 |
from pipeline.custom_pipelines import FluxPriorReduxPipeline, FluxControlNetImg2ImgPipeline, FluxImg2ImgPipeline
|
| 32 |
-
from diffusers import FluxPipeline, DiffusionPipeline, EulerAncestralDiscreteScheduler, FluxTransformer2DModel
|
| 33 |
from diffusers.models.controlnets.controlnet_flux import FluxMultiControlNetModel, FluxControlNetModel
|
| 34 |
from diffusers.schedulers import FlowMatchHeunDiscreteScheduler
|
| 35 |
from huggingface_hub import hf_hub_download
|
|
|
|
| 29 |
from models.llm.llm import load_llm_model, get_llm_response
|
| 30 |
|
| 31 |
from pipeline.custom_pipelines import FluxPriorReduxPipeline, FluxControlNetImg2ImgPipeline, FluxImg2ImgPipeline
|
| 32 |
+
from diffusers import FluxPipeline, DiffusionPipeline, EulerAncestralDiscreteScheduler, FluxTransformer2DModel, AutoencoderTiny
|
| 33 |
from diffusers.models.controlnets.controlnet_flux import FluxMultiControlNetModel, FluxControlNetModel
|
| 34 |
from diffusers.schedulers import FlowMatchHeunDiscreteScheduler
|
| 35 |
from huggingface_hub import hf_hub_download
|