fix: add weights_only=True to all torch.load call sites - #2622

fix: add weights_only=True to all torch.load call sites#2622

tridao merged 1 commit into

Dao-AILab:mainDao-AILab/flash-attention:mainfrom

aryanputta:fix/weights-only-torch-loadaryanputta/flash-attention:fix/weights-only-torch-loadCopy head branch name to clipboard