Skip to content

Commit a2486a5

Browse files
RyanJDickmaryhipp
authored andcommitted
Remove unused prediction_type and upcast_attention from from_single_file(...) calls.
1 parent 07ab116 commit a2486a5

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

invokeai/backend/model_manager/load/model_loaders/stable_diffusion.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -117,8 +117,6 @@ def _load_from_singlefile(
117117
load_class = load_classes[config.base][config.variant]
118118
except KeyError as e:
119119
raise Exception(f"No diffusers pipeline known for base={config.base}, variant={config.variant}") from e
120-
prediction_type = config.prediction_type.value
121-
upcast_attention = config.upcast_attention
122120

123121
# Without SilenceWarnings we get log messages like this:
124122
# site-packages/huggingface_hub/file_download.py:1132: FutureWarning: `resume_download` is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use `force_download=True`.
@@ -129,12 +127,7 @@ def _load_from_singlefile(
129127
# ['text_model.embeddings.position_ids']
130128

131129
with SilenceWarnings():
132-
pipeline = load_class.from_single_file(
133-
config.path,
134-
torch_dtype=self._torch_dtype,
135-
prediction_type=prediction_type,
136-
upcast_attention=upcast_attention,
137-
)
130+
pipeline = load_class.from_single_file(config.path, torch_dtype=self._torch_dtype)
138131

139132
if not submodel_type:
140133
return pipeline

0 commit comments

Comments
 (0)