Spaces:

KeerthiVM
/

SkinGPT

Sleeping

KeerthiVM commited on May 7

Commit

3b2ce40

1 Parent(s): 7f3d8d1

fix added

Files changed (1) hide show

SkinGPT.py CHANGED Viewed

@@ -71,7 +71,6 @@ class Blip2QFormer(nn.Module):
             output_attentions=True,
             return_dict=True
         )
-        plot_attention(outputs.attentions[-1][:, :, :32, 32:])
         return outputs.last_hidden_state[:, :self.num_query_tokens]

             output_attentions=True,
             return_dict=True
         )
         return outputs.last_hidden_state[:, :self.num_query_tokens]