Update app.py

app.py

@@ -49,6 +49,11 @@ model, _, _ = load_model(config, ckpt_path,
                          inject_lora=False, 
                          lora_scale=None, 
                          )
+state_dict = torch.load(adapter_ckpt, map_location="cpu")
+if "state_dict" in list(state_dict.keys()):
+    state_dict = state_dict["state_dict"]
+model.adapter.load_state_dict(state_dict, strict=True)
+
 ddim_sampler = DDIMSampler(model)
 
 def sample_denoising_batch(model, noise_shape, condition, *args,