From e8814e59e3eb76b5330e88c8ae53a22e168c6895 Mon Sep 17 00:00:00 2001 From: HfCloud Date: Sun, 2 Jul 2023 16:57:43 +0800 Subject: [PATCH] update to v1.2 --- bot.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bot.py b/bot.py index ef7f642..b3ff2e6 100644 --- a/bot.py +++ b/bot.py @@ -59,7 +59,7 @@ def run_safety_nochecker(image, device, dtype): sd_args = { "pretrained_model_name_or_path" : config_json["Diffusion"]["Diffusion-Model"], - "torch_dtype" : (torch.float16 if config_json.get("UseFP16", True) else torch.float32) + "torch_dtype" : (torch.float16 if config_json["Diffusion"].get("UseFP16", True) else torch.float32) } sd_pipe = StableDiffusionPipeline.from_pretrained(**sd_args)