VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-06 13:47:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2267 from kohya-ss/fix-github-actions-error

Browse Source 
fix: `str is not "no"` to `str != "no"`
This commit is contained in:
Kohya S.
2026-02-16 08:01:20 +09:00
committed by GitHub
parent 573a7fa06c ef051427df
commit 3265f2edfb
1 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
library/deepspeed_utils.py
View File

@@ -128,7 +128,7 @@ def prepare_deepspeed_model(args: argparse.Namespace, **models):
self.models = torch.nn.ModuleDict()
wrap_model_forward_with_torch_autocast = args.mixed_precision is not "no"
wrap_model_forward_with_torch_autocast = args.mixed_precision != "no"
for key, model in kw_models.items():
if isinstance(model, list):
@@ -175,6 +175,5 @@ def prepare_deepspeed_model(args: argparse.Namespace, **models):
def get_models(self):
return self.models
ds_model = DeepSpeedWrapper(**models)
return ds_model