update app.py

app.py

@@ -134,7 +134,8 @@ def process(prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_step
 
  # set up cfg
  uc_local_control = local_control
- uc_global_control = get_unconditional_global(global_control)
+ uc_global_control = {k:torch.zeros_like(v) for k,v in global_control.items()}
+
  cond = {
  "local_control": [local_control], 
  "c_crossattn": [model.get_learned_conditioning([prompt + ', ' + a_prompt] * num_samples)], 
@@ -192,7 +193,7 @@ define model
 config_file = "configs/anycontrol.yaml"
 model_file = "ckpts/anycontrol_15.ckpt"
 model = create_model(config_file).cpu()
-model.load_state_dict(load_state_dict(model_file, location='cuda'))
+model.load_state_dict(load_state_dict(model_file, location='cpu'))
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)