compvis / attribute-control Goto Github PK

[2024-04-02 16:40:02,359][__main__][INFO] - Starting optimization...
Optimizing:   0%|                                                                                                                                                                                              | 0/1000 [02:46<?, ?it/s]
Error executing job with overrides: ['device=cuda:0', 'model=sd15', 'prompts=people/age']
Traceback (most recent call last):
  File "/home/yinzijin/attribute-control/learn_delta.py", line 81, in main
    eps_p = model.predict_eps(prompts_embedded['prompt_positive'], x_t, t_relative).unsqueeze(0)
  File "/home/yinzijin/attribute-control/attribute_control/model/model.py", line 223, in predict_eps
    return self._get_eps_pred(t, start_sample, self.pipe.unet(start_sample, t, encoder_hidden_states=self._get_pipe_kwargs(embs, embs_neg=None, start_sample=None)['prompt_embeds']).sample)
  File "/home/yinzijin/attribute-control/attribute_control/model/model.py", line 164, in _get_eps_pred
    return (sample - beta_prod_t ** (0.5) * model_output) / alpha_prod_t ** (0.5)
RuntimeError: The size of tensor a (2) must match the size of tensor b (64) at non-singleton dimension 3

torch                     2.2.0+cu121
torchvision               0.17.0+cu121
diffusers                 0.27.2
accelerate                0.28.0