Update hieradet.py
Not used head_dim = dim_out // num_heads self.scale = head_dim**-0.5 F.scaled_dot_product_attention takes care of this automatically.
A
Arun committed
6ec8560436159db49fa8d209267d6feaf6d8c43b
Parent: 511199d
Committed by GitHub <noreply@github.com>
on 8/7/2024, 6:05:46 AM