Spaces:

Gradio-Blocks
/

clip-guided-faces

Runtime error

sxela commited on Jun 13, 2022

Commit

a88186d

1 Parent(s): e5a1104

bump model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -107,7 +107,7 @@ def inference(text, init_image, skip_timesteps, clip_guidance_scale, tv_scale, r
     print('Using fp16: ',model_config['use_fp16'])
     print('Using device:', device)
     model, diffusion = create_model_and_diffusion(**model_config)
-    model.load_state_dict(torch.load('256x256_openai_comics_faces_by_alex_spirin_084000.pt', map_location='cpu'))
     model.requires_grad_(False).eval().to(device).float()
     for name, param in model.named_parameters():
         if 'qkv' in name or 'norm' in name or 'proj' in name:

     print('Using fp16: ',model_config['use_fp16'])
     print('Using device:', device)
     model, diffusion = create_model_and_diffusion(**model_config)
+    model.load_state_dict(torch.load('256x256_openai_comics_faces_v2.by_alex_spirin_114k.pt', map_location='cpu'))
     model.requires_grad_(False).eval().to(device).float()
     for name, param in model.named_parameters():
         if 'qkv' in name or 'norm' in name or 'proj' in name: