Commit
•
b51595d
1
Parent(s):
a5ebb54
Refactor model loading in load_models.py to use ControlNetModel instead of FluxControlNetModel
Browse files
tabs/images/load_models.py
CHANGED
|
@@ -3,6 +3,7 @@ from diffusers import (
|
|
| 3 |
AutoPipelineForText2Image,
|
| 4 |
AutoencoderKL,
|
| 5 |
FluxControlNetModel,
|
|
|
|
| 6 |
FluxMultiControlNetModel,
|
| 7 |
)
|
| 8 |
from diffusers.schedulers import *
|
|
|
|
| 3 |
AutoPipelineForText2Image,
|
| 4 |
AutoencoderKL,
|
| 5 |
FluxControlNetModel,
|
| 6 |
+
ControlNetModel,
|
| 7 |
FluxMultiControlNetModel,
|
| 8 |
)
|
| 9 |
from diffusers.schedulers import *
|