search

IrisRainbowNeko / HCP-Diffusion

A universal Stable-Diffusion toolbox
Apache License 2.0
899 stars 75 forks source link

重构了sd2difusers模块部分函数 #29

Closed waterminer closed 1 year ago

waterminer commented 1 year ago

修改内容

args_class模块

我创建了一个args_class模块，它放在utils文件夹里 这个模块目前只做了sd2diffusers需要的参数，将他包装在Sd2diffusers_convert_args类里 它的初始化函数，可以选择把args传进去来初始化

sd2diffusers模块

将sd2diffusers中的convert_ckpt和sd_vae_to_diffuser改成了只要传几个必要参数就行 （不然每次调用都得自己实现一整个args类然后传进去，那样太蠢了）

convert_ckp有四个形参，分别是checkpoint_path original_config_file dump_path以及一个带有默认值的args args指定需要Sd2diffusers_convert_args类型

sd_vae_to_diffuser有四个形参，分别是vae_pt_path dump_path original_config_file以及一个带有默认值的args args指定需要Sd2diffusers_convert_args类型

waterminer commented 1 year ago

什么俩初始化函数 写出BUG了（写JAVA那种OOP代码写的，啥都想重载） 改成一个了 （然后不懂PR关了开开了关，丢人丢到姥姥家）