I think you are right that it should be PyTorch's issue setting the parameter. Could you print the fx graph, and/or try exporting with torch.export.export()? Contributor Author yuanyao-nv commented Feb 5, 2024
diff --git a/.cloudbuild.yaml b/.cloudbuild.yaml index 1412ab92b62..c17c7753415 100644 --- a/.cloudbuild.yaml +++ b/.cloudbuild.yaml @@ -192,10 +192,10 @@ steps: args: ['pull', 'gcr.io/cloudsql-docker/gce-proxy:1.14'] id: 'pullCloudsqlProxy' - name: 'gcr.io/cloud-...
Support torch>=2.6 in word_language_model example (#1347) May 18, 2025 runtime.txt Adding runtime.txt for Netlify Apr 28, 2022 utils.sh Respect each example requirements and use uv (#1330) Apr 26, 2025 PyTorch Examples https://pytorch.org/examples/ ...