Skip to content

Commit 52ab514

Browse files
committed
Silence W0237 pylint warning
1 parent b5536b5 commit 52ab514

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

tools/accuracy_checker/accuracy_checker/evaluators/custom_evaluators/ldm_super_resolution_evaluator.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -94,13 +94,13 @@ def load_models(self, model_info, launcher, log=False):
9494
if log:
9595
self.print_input_output_info()
9696

97-
def load_model(self, network_info, launcher):
98-
model, weights = self.automatic_model_search(network_info)
97+
def load_model(self, network_list, launcher):
98+
model, weights = self.automatic_model_search(network_list)
9999
if weights:
100100
network = launcher.read_network(str(model), str(weights))
101101
else:
102102
network = launcher.read_network(str(model), None)
103-
setattr(self, f"{network_info['name']}_model", network)
103+
setattr(self, f"{network_list['name']}_model", network)
104104

105105
def print_input_output_info(self):
106106
model_parts = ("unet", "vqvae")

tools/accuracy_checker/accuracy_checker/evaluators/custom_evaluators/stable_diffusion_evaluator.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -89,13 +89,13 @@ def load_models(self, model_info, launcher, log=False):
8989
self.print_input_output_info()
9090

9191

92-
def load_model(self, network_info, launcher):
93-
model, weights = self.automatic_model_search(network_info)
92+
def load_model(self, network_list, launcher):
93+
model, weights = self.automatic_model_search(network_list)
9494
if weights:
9595
network = launcher.read_network(str(model), str(weights))
9696
else:
9797
network = launcher.read_network(str(model), None)
98-
setattr(self, f"{network_info['name']}_model", network)
98+
setattr(self, f"{network_list['name']}_model", network)
9999

100100
def print_input_output_info(self):
101101
model_parts = ("text_encoder", "unet", "vae_decoder", "vae_encoder")

0 commit comments

Comments
 (0)