Yinhong Liu
commited on
Commit
·
76c9b16
1
Parent(s):
bea27d1
fix sd3 pipeline
Browse files- sid/__init__.py +3 -3
sid/__init__.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
from typing import TYPE_CHECKING
|
| 2 |
|
| 3 |
-
from
|
| 4 |
DIFFUSERS_SLOW_IMPORT,
|
| 5 |
OptionalDependencyNotAvailable,
|
| 6 |
_LazyModule,
|
|
@@ -19,7 +19,7 @@ try:
|
|
| 19 |
if not (is_transformers_available() and is_torch_available()):
|
| 20 |
raise OptionalDependencyNotAvailable()
|
| 21 |
except OptionalDependencyNotAvailable:
|
| 22 |
-
from
|
| 23 |
|
| 24 |
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
| 25 |
else:
|
|
@@ -32,7 +32,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
|
| 32 |
if not (is_transformers_available() and is_torch_available()):
|
| 33 |
raise OptionalDependencyNotAvailable()
|
| 34 |
except OptionalDependencyNotAvailable:
|
| 35 |
-
from
|
| 36 |
else:
|
| 37 |
from .pipeline_sid_sd3 import SiDSD3Pipeline
|
| 38 |
from .pipeline_sid_flux import SiDFluxPipeline
|
|
|
|
| 1 |
from typing import TYPE_CHECKING
|
| 2 |
|
| 3 |
+
from diffusers.utils import (
|
| 4 |
DIFFUSERS_SLOW_IMPORT,
|
| 5 |
OptionalDependencyNotAvailable,
|
| 6 |
_LazyModule,
|
|
|
|
| 19 |
if not (is_transformers_available() and is_torch_available()):
|
| 20 |
raise OptionalDependencyNotAvailable()
|
| 21 |
except OptionalDependencyNotAvailable:
|
| 22 |
+
from diffusers.utils import dummy_torch_and_transformers_objects # noqa F403
|
| 23 |
|
| 24 |
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
| 25 |
else:
|
|
|
|
| 32 |
if not (is_transformers_available() and is_torch_available()):
|
| 33 |
raise OptionalDependencyNotAvailable()
|
| 34 |
except OptionalDependencyNotAvailable:
|
| 35 |
+
from diffusers.utils.dummy_torch_and_transformers_objects import * # noqa F403
|
| 36 |
else:
|
| 37 |
from .pipeline_sid_sd3 import SiDSD3Pipeline
|
| 38 |
from .pipeline_sid_flux import SiDFluxPipeline
|