Fix for weights-only load #1228
Dr.CI classification results
{"FAILED":[],"FLAKY":[],"BROKEN_TRUNK":[{"workflowId":11695903019,"workflowUniqueId":89543087,"id":32572116194,"runnerName":"i-0d94807f8ad515e98","authorEmail":"drisspguessous@gmail.com","name":"Run Regression Tests / test (CPU Nightly, linux.4xlarge, --pre torch --index-url https://download.pytorch.org/whl/nightl... / linux-job","jobName":"test (CPU Nightly, linux.4xlarge, --pre torch --index-url https://download.pytorch.org/whl/nightl... / linux-job","conclusion":"failure","completed_at":"2024-11-06T02:19:01.000000000Z","html_url":"https://github.com/pytorch/ao/actions/runs/11695903019/job/32572116194","head_branch":"drisspg/stack/19","pr_number":1228,"head_sha":"21b3bc2908bf5b4e3e2472142f9eff317bf72984","head_sha_timestamp":"2024-11-06T02:07:27.000000000Z","failure_captures":[],"failure_lines":[],"failure_context":[],"time":"2024-11-06T02:08:30.000000000Z"},{"workflowId":11695903019,"workflowUniqueId":89543087,"id":32572117399,"runnerName":"i-0220bbe991fb40dc7","authorEmail":"drisspguessous@gmail.com","name":"Run Regression Tests / test (CUDA Nightly, linux.g5.12xlarge.nvidia.gpu, --pre torch --index-url https://download.pytorc... / linux-job","jobName":"test (CUDA Nightly, linux.g5.12xlarge.nvidia.gpu, --pre torch --index-url https://download.pytorc... / linux-job","conclusion":"failure","completed_at":"2024-11-06T03:14:41.000000000Z","html_url":"https://github.com/pytorch/ao/actions/runs/11695903019/job/32572117399","head_branch":"drisspg/stack/19","pr_number":1228,"head_sha":"21b3bc2908bf5b4e3e2472142f9eff317bf72984","head_sha_timestamp":"2024-11-06T02:07:27.000000000Z","failure_captures":[],"failure_lines":[],"failure_context":[],"time":"2024-11-06T02:08:33.000000000Z"}],"UNSTABLE":[]}