Home Creators Posts Import Register Favorites Logout

Content

The fourth and final checkpoint of the faces model is available for early access. 

You can get it here: https://drive.google.com/file/d/1o06GiaQGNtuVa9NqlH2lwM_lIX0w42eO/view?usp=sharing

and plug it directly in here (just change the custom model path parameter in model settings):

https://github.com/Sxela/DiscoDiffusion-Warp/blob/main/Disco_Diffusion_v5_2_Warp_custom_model.ipynb

More info on how to use a custom model is here: https://www.patreon.com/posts/66291282


Comments

Felipe González

Hi, that is great. Any specific custom model settings? I got this error when I tried to run this custom model: 203 model, diffusion = create_model_and_diffusion(**model_config) 204 if diffusion_model == 'custom': --> 205 model.load_state_dict(torch.load(custom_path, map_location='cpu')) 206 else: 207 model.load_state_dict(torch.load(f'{model_path}/{diffusion_model}.pt', map_location='cpu'))