Load Anything-v4 models

model.py

@@ -67,18 +67,21 @@ class Model:
         config.model.params.cond_stage_config.params.device = device
 
         base_model_file = "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt"
+        base_model_file_anything = "https://huggingface.co/andite/anything-v4.0/resolve/main/anything-v4.0-pruned.ckpt"
         sketch_adapter_file = "https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_sketch_sd14v1.pth"
         pose_adapter_file = "https://huggingface.co/TencentARC/T2I-Adapter/resolve/main/models/t2iadapter_keypose_sd14v1.pth" 
         pidinet_file = model_path+"table5_pidinet.pth"
         clip_file = "https://huggingface.co/openai/clip-vit-large-patch14/resolve/main/*"
         
         subprocess.run(shlex.split(f'wget {base_model_file} -O models/sd-v1-4.ckpt'))
+        subprocess.run(shlex.split(f'wget {base_model_file_anything} -O models/anything-v4.0-pruned.ckpt'))
         subprocess.run(shlex.split(f'wget {sketch_adapter_file} -O models/t2iadapter_sketch_sd14v1.pth'))
         subprocess.run(shlex.split(f'wget {pose_adapter_file} -O models/t2iadapter_keypose_sd14v1.pth'))
         subprocess.run(shlex.split(f'wget {pidinet_file} -O models/table5_pidinet.pth'))
 
         
         self.model = load_model_from_config(config, "models/sd-v1-4.ckpt").to(device)
+        self.model_anything = load_model_from_config(config, "models/anything-v4.0-pruned.ckpt").to(device)
         current_base = 'sd-v1-4.ckpt'
         self.model_ad_sketch = Adapter(channels=[320, 640, 1280, 1280][:4], nums_rb=2, ksize=1, sk=True, use_conv=False).to(device)
         self.model_ad_sketch.load_state_dict(torch.load("models/t2iadapter_sketch_sd14v1.pth"))
@@ -87,6 +90,7 @@ class Model:
         net_G.load_state_dict({k.replace('module.',''):v for k, v in ckp.items()})
         net_G.to(device)
         self.sampler= PLMSSampler(self.model)
+        self.sampler_anything= PLMSSampler(self.model_anything)
         save_memory=True
 
         self.model_ad_pose = Adapter(cin=int(3*64),channels=[320, 640, 1280, 1280][:4], nums_rb=2, ksize=1, sk=True, use_conv=False).to(device)