Big-Data-Lab
/
WeChatBot
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update to v1.2

This commit is contained in:
HfCloud 2023-07-02 16:57:43 +08:00
parent adf34e91e9
commit e8814e59e3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bot.py
View File

@ -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)