Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
988c8222
Commit
988c8222
authored
Nov 03, 2022
by
Patrick von Platen
Browse files
fix copies
parent
74821781
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
75 additions
and
0 deletions
+75
-0
src/diffusers/utils/dummy_torch_and_accelerate_objects.py
src/diffusers/utils/dummy_torch_and_accelerate_objects.py
+75
-0
No files found.
src/diffusers/utils/dummy_torch_and_accelerate_objects.py
View file @
988c8222
...
@@ -34,6 +34,21 @@ class AutoencoderKL(metaclass=DummyObject):
...
@@ -34,6 +34,21 @@ class AutoencoderKL(metaclass=DummyObject):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
class
Transformer2DModel
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
,
"accelerate"
]
def
__init__
(
self
,
*
args
,
**
kwargs
):
requires_backends
(
self
,
[
"torch"
,
"accelerate"
])
@
classmethod
def
from_config
(
cls
,
*
args
,
**
kwargs
):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
@
classmethod
def
from_pretrained
(
cls
,
*
args
,
**
kwargs
):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
class
UNet1DModel
(
metaclass
=
DummyObject
):
class
UNet1DModel
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
,
"accelerate"
]
_backends
=
[
"torch"
,
"accelerate"
]
...
@@ -227,6 +242,21 @@ class PNDMPipeline(metaclass=DummyObject):
...
@@ -227,6 +242,21 @@ class PNDMPipeline(metaclass=DummyObject):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
class
RePaintPipeline
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
,
"accelerate"
]
def
__init__
(
self
,
*
args
,
**
kwargs
):
requires_backends
(
self
,
[
"torch"
,
"accelerate"
])
@
classmethod
def
from_config
(
cls
,
*
args
,
**
kwargs
):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
@
classmethod
def
from_pretrained
(
cls
,
*
args
,
**
kwargs
):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
class
ScoreSdeVePipeline
(
metaclass
=
DummyObject
):
class
ScoreSdeVePipeline
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
,
"accelerate"
]
_backends
=
[
"torch"
,
"accelerate"
]
...
@@ -242,6 +272,21 @@ class ScoreSdeVePipeline(metaclass=DummyObject):
...
@@ -242,6 +272,21 @@ class ScoreSdeVePipeline(metaclass=DummyObject):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
class
VQDiffusionPipeline
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
,
"accelerate"
]
def
__init__
(
self
,
*
args
,
**
kwargs
):
requires_backends
(
self
,
[
"torch"
,
"accelerate"
])
@
classmethod
def
from_config
(
cls
,
*
args
,
**
kwargs
):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
@
classmethod
def
from_pretrained
(
cls
,
*
args
,
**
kwargs
):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
class
DDIMScheduler
(
metaclass
=
DummyObject
):
class
DDIMScheduler
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
,
"accelerate"
]
_backends
=
[
"torch"
,
"accelerate"
]
...
@@ -347,6 +392,21 @@ class PNDMScheduler(metaclass=DummyObject):
...
@@ -347,6 +392,21 @@ class PNDMScheduler(metaclass=DummyObject):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
class
RePaintScheduler
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
,
"accelerate"
]
def
__init__
(
self
,
*
args
,
**
kwargs
):
requires_backends
(
self
,
[
"torch"
,
"accelerate"
])
@
classmethod
def
from_config
(
cls
,
*
args
,
**
kwargs
):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
@
classmethod
def
from_pretrained
(
cls
,
*
args
,
**
kwargs
):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
class
SchedulerMixin
(
metaclass
=
DummyObject
):
class
SchedulerMixin
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
,
"accelerate"
]
_backends
=
[
"torch"
,
"accelerate"
]
...
@@ -377,6 +437,21 @@ class ScoreSdeVeScheduler(metaclass=DummyObject):
...
@@ -377,6 +437,21 @@ class ScoreSdeVeScheduler(metaclass=DummyObject):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
class
VQDiffusionScheduler
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
,
"accelerate"
]
def
__init__
(
self
,
*
args
,
**
kwargs
):
requires_backends
(
self
,
[
"torch"
,
"accelerate"
])
@
classmethod
def
from_config
(
cls
,
*
args
,
**
kwargs
):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
@
classmethod
def
from_pretrained
(
cls
,
*
args
,
**
kwargs
):
requires_backends
(
cls
,
[
"torch"
,
"accelerate"
])
class
EMAModel
(
metaclass
=
DummyObject
):
class
EMAModel
(
metaclass
=
DummyObject
):
_backends
=
[
"torch"
,
"accelerate"
]
_backends
=
[
"torch"
,
"accelerate"
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment