mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix #1935 , torch.amp.autocast -> torch.autocast (namespace issue for 1.10 pt compat)

vit_packed
Ross Wightman 2023-08-30 15:02:20 -07:00
parent 9260cf517d
commit 983310d6a2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/efficientvit_mit.py
View File

@ -260,7 +260,7 @@ class LiteMSA(nn.Module):
v = F.pad(v, (0, 1), mode="constant", value=1.)
if not torch.jit.is_scripting():
with torch.amp.autocast(device_type=v.device.type, enabled=False):
with torch.autocast(device_type=v.device.type, enabled=False):
out = self._attn(q, k, v)
else:
out = self._attn(q, k, v)