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

fix pipeline dtype

Browse Source
This commit is contained in:
Isotr0py
2023-07-27 23:16:58 +08:00
parent eec6aaddda
commit 50544b7805
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/sdxl_train_util.py
View File

@@ -74,7 +74,7 @@ def _load_target_model(args: argparse.Namespace, model_version: str, weight_dtyp
print(f"load Diffusers pretrained models: {name_or_path}, variant={variant}") print(f"load Diffusers pretrained models: {name_or_path}, variant={variant}")
try: try:
try: try:
pipe = StableDiffusionXLPipeline.from_pretrained(name_or_path, variant=variant, tokenizer=None) pipe = StableDiffusionXLPipeline.from_pretrained(name_or_path, torch_dtype=weight_dtype, variant=variant, tokenizer=None)
except EnvironmentError as ex: except EnvironmentError as ex:
if variant is not None: if variant is not None:
print("try to load fp32 model") print("try to load fp32 model")