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
8dc93ad3
Unverified
Commit
8dc93ad3
authored
Sep 14, 2023
by
Patrick von Platen
Committed by
GitHub
Sep 14, 2023
Browse files
[Import] Don't force transformers to be installed (#5035)
* [Import] Don't force transformers to be installed * make style
parent
e2033d2d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
5 deletions
+5
-5
src/diffusers/loaders.py
src/diffusers/loaders.py
+5
-5
No files found.
src/diffusers/loaders.py
View file @
8dc93ad3
...
@@ -42,7 +42,7 @@ from .utils.import_utils import BACKENDS_MAPPING
...
@@ -42,7 +42,7 @@ from .utils.import_utils import BACKENDS_MAPPING
if
is_transformers_available
():
if
is_transformers_available
():
from
transformers
import
CLIPTextModel
,
CLIPTextModelWithProjection
,
PreTrainedModel
,
PreTrainedTokenizer
from
transformers
import
CLIPTextModel
,
CLIPTextModelWithProjection
if
is_accelerate_available
():
if
is_accelerate_available
():
from
accelerate
import
init_empty_weights
from
accelerate
import
init_empty_weights
...
@@ -628,7 +628,7 @@ class TextualInversionLoaderMixin:
...
@@ -628,7 +628,7 @@ class TextualInversionLoaderMixin:
Load textual inversion tokens and embeddings to the tokenizer and text encoder.
Load textual inversion tokens and embeddings to the tokenizer and text encoder.
"""
"""
def
maybe_convert_prompt
(
self
,
prompt
:
Union
[
str
,
List
[
str
]],
tokenizer
:
"PreTrainedTokenizer"
):
def
maybe_convert_prompt
(
self
,
prompt
:
Union
[
str
,
List
[
str
]],
tokenizer
:
"PreTrainedTokenizer"
):
# noqa: F821
r
"""
r
"""
Processes prompts that include a special token corresponding to a multi-vector textual inversion embedding to
Processes prompts that include a special token corresponding to a multi-vector textual inversion embedding to
be replaced with multiple special tokens each corresponding to one of the vectors. If the prompt has no textual
be replaced with multiple special tokens each corresponding to one of the vectors. If the prompt has no textual
...
@@ -655,7 +655,7 @@ class TextualInversionLoaderMixin:
...
@@ -655,7 +655,7 @@ class TextualInversionLoaderMixin:
return
prompts
return
prompts
def
_maybe_convert_prompt
(
self
,
prompt
:
str
,
tokenizer
:
"PreTrainedTokenizer"
):
def
_maybe_convert_prompt
(
self
,
prompt
:
str
,
tokenizer
:
"PreTrainedTokenizer"
):
# noqa: F821
r
"""
r
"""
Maybe convert a prompt into a "multi vector"-compatible prompt. If the prompt includes a token that corresponds
Maybe convert a prompt into a "multi vector"-compatible prompt. If the prompt includes a token that corresponds
to a multi-vector textual inversion embedding, this function will process the prompt so that the special token
to a multi-vector textual inversion embedding, this function will process the prompt so that the special token
...
@@ -689,8 +689,8 @@ class TextualInversionLoaderMixin:
...
@@ -689,8 +689,8 @@ class TextualInversionLoaderMixin:
self
,
self
,
pretrained_model_name_or_path
:
Union
[
str
,
List
[
str
],
Dict
[
str
,
torch
.
Tensor
],
List
[
Dict
[
str
,
torch
.
Tensor
]]],
pretrained_model_name_or_path
:
Union
[
str
,
List
[
str
],
Dict
[
str
,
torch
.
Tensor
],
List
[
Dict
[
str
,
torch
.
Tensor
]]],
token
:
Optional
[
Union
[
str
,
List
[
str
]]]
=
None
,
token
:
Optional
[
Union
[
str
,
List
[
str
]]]
=
None
,
tokenizer
:
Optional
[
PreTrainedTokenizer
]
=
None
,
tokenizer
:
Optional
[
"
PreTrainedTokenizer
"
]
=
None
,
# noqa: F821
text_encoder
:
Optional
[
PreTrainedModel
]
=
None
,
text_encoder
:
Optional
[
"
PreTrainedModel
"
]
=
None
,
# noqa: F821
**
kwargs
,
**
kwargs
,
):
):
r
"""
r
"""
...
...
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