Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-10-23 08:46:05 -04:00
parent 3cc78bd3ca
commit 6d4f2dfd7e
5 changed files with 7 additions and 6 deletions
+1 -1
View File
@@ -44,7 +44,7 @@ class GeneralizedLinearAttention(Attention):
self.to_q_ = get_none_linear_projection(query_dim, mid_dim, **kwargs)
self.to_k_ = get_none_linear_projection(query_dim, mid_dim, **kwargs)
def forward(
def forward( # pylint: disable=unused-argument
self,
hidden_states,
encoder_hidden_states=None,
+1 -1
View File
@@ -86,7 +86,7 @@ class LinFusion(ModelMixin, ConfigMixin):
if load_pretrained:
# Load from pretrained
if not pretrained_model_name_or_path:
pipe_name_path = pipe_name_path or pipeline._internal_dict._name_or_path
pipe_name_path = pipe_name_path or pipeline._internal_dict._name_or_path # pylint: disable=protected-access
pretrained_model_name_or_path = model_dict.get(pipe_name_path, None)
if pretrained_model_name_or_path:
print(
+1 -1
View File
@@ -191,7 +191,7 @@ class DiffusionSampler:
if 'lower_order_final' in self.config:
self.config['lower_order_final'] = shared.opts.schedulers_use_loworder
if 'solver_order' in self.config and int(shared.opts.schedulers_solver_order) > 0:
self.config['solver_order'] = (shared.opts.schedulers_solver_order)
self.config['solver_order'] = int(shared.opts.schedulers_solver_order)
if 'predict_x0' in self.config:
self.config['solver_type'] = shared.opts.uni_pc_variant
if 'beta_start' in self.config and shared.opts.schedulers_beta_start > 0: