SAM2 AMG load_fast fix

[cpuhrsch]

Resolves compatibility with furious mode and load-fast.

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/1374

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 9 New Failures

As of commit 0e11119 with merge base 8a51e1a ():

NEW FAILURES - The following jobs have failed:

• Run Float8 Tests / test (SM-89, linux.g6.4xlarge.experimental.nvidia.gpu, --pre torch --index-url https://download.p... / linux-job (gh)
  ##[error]fatal: couldn't find remote ref refs/pull/1374/merge
• Run Regression Tests / test (CPU 2.3, linux.4xlarge, torch==2.3.0 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  ##[error]fatal: couldn't find remote ref refs/pull/1374/merge
• Run Regression Tests / test (CPU 2.4, linux.4xlarge, torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  ##[error]fatal: couldn't find remote ref refs/pull/1374/merge
• Run Regression Tests / test (CPU 2.5.1, linux.4xlarge, torch==2.5.1 --index-url https://download.pytorch.org/whl/cpu, cpu) / linux-job (gh)
  ##[error]fatal: couldn't find remote ref refs/pull/1374/merge
• Run Regression Tests / test (CUDA 2.3, linux.g5.12xlarge.nvidia.gpu, torch==2.3.0, cuda, 12.1) / linux-job (gh)
  ##[error]fatal: couldn't find remote ref refs/pull/1374/merge
• Run Regression Tests / test (CUDA 2.4, linux.g5.12xlarge.nvidia.gpu, torch==2.4.0, cuda, 12.1) / linux-job (gh)
  ##[error]fatal: couldn't find remote ref refs/pull/1374/merge
• Run Regression Tests / test (CUDA 2.5.1, linux.g5.12xlarge.nvidia.gpu, torch==2.5.1 --index-url https://download.pytorch... / linux-job (gh)
  ##[error]fatal: couldn't find remote ref refs/pull/1374/merge
• Run Regression Tests / test-nightly (CPU Nightly, linux.4xlarge, --pre torch --index-url https://download.pytorch.org/wh... / linux-job (gh)
  ##[error]fatal: couldn't find remote ref refs/pull/1374/merge
• Run Regression Tests / test-nightly (CUDA Nightly, linux.g5.12xlarge.nvidia.gpu, --pre torch --index-url https://downloa... / linux-job (gh)
  ##[error]fatal: couldn't find remote ref refs/pull/1374/merge

This comment was automatically generated by Dr. CI and updates every 15 minutes.