nodes/space-nuko-ComfyUI-Disco-Diffusion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update Log Messages

Browse Source
This commit is contained in:
WAS
2023-05-17 22:34:53 -07:00
committed by GitHub
parent 235a0fc0fd
commit e811667cc7
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
model_settings.py
+5 -5
View File
@@ -130,24 +130,24 @@ class ModelSettings:
model_local_path = os.path.join(self.model_path, model_filename)
if os.path.exists(model_local_path):
if self.check_model_SHA:
print(f"Checking {model_filename} SHA256 Hash...")
#print(f"Checking {model_filename} SHA256 Hash...")
if self.check_sha_hash(diffusion_model_name, model_local_path):
print(f'{diffusion_model_name} SHA256 matches.')
#print(f'{diffusion_model_name} SHA256 matches.')
diff_model_map[diffusion_model_name]['downloaded'] = True
else:
print(f"{diffusion_model_name} SHA doesn't match. Will redownload it...")
os.remove(model_local_path)
return self.download_model(diffusion_model_name)
print(f"{model_filename} \033[32malready downloaded.\033[0;0m")
#print(f"{model_filename} \033[32malready downloaded.\033[0;0m")
else:
for model_uri in diff_model_map[diffusion_model_name]['uri_list']:
print(f"Downloading {diffusion_model_name} from: {model_uri}")
disco_utils.pyget(model_uri, self.model_path)
if os.path.exists(model_local_path):
if self.check_model_SHA:
print(f"Checking {model_filename} SHA256 Hash...")
#print(f"Checking {model_filename} SHA256 Hash...")
if self.check_sha_hash(diffusion_model_name, model_local_path):
print(f'{diffusion_model_name} SHA256 matches.')
#print(f'{diffusion_model_name} SHA256 matches.')
diff_model_map[diffusion_model_name]['downloaded'] = True
print(f"{diffusion_model_name} \033[32mdownloaded successfully.\033[0;0m")
return