Update torch for image captioning

Amar3tto · Amar3tto · commit 7dd82b3f6f97 · 2026-01-22T19:17:26.000+04:00
diff --git a/sdks/python/apache_beam/ml/inference/pytorch_image_captioning_requirements.txt b/sdks/python/apache_beam/ml/inference/pytorch_image_captioning_requirements.txt
@@ -15,8 +15,7 @@
 # limitations under the License.
 #
 
-torch>=2.2.0,<2.8.0
-torchvision>=0.17.0,<0.21.0
+torch>=2.6.0,<2.8.0
 Pillow>=10.0.0
 numpy>=1.25.0
 transformers>=4.41.0,<5.0.0

Original file line number	Diff line number	Diff line change
`@@ -15,8 +15,7 @@`
`15`	`15`	`# limitations under the License.`
`16`	`16`	`#`
`17`	`17`
`18`		`-torch>=2.2.0,<2.8.0`
`19`		`-torchvision>=0.17.0,<0.21.0`
	`18`	`+torch>=2.6.0,<2.8.0`
`20`	`19`	`Pillow>=10.0.0`
`21`	`20`	`numpy>=1.25.0`
`22`	`21`	`transformers>=4.41.0,<5.0.0`