make torch a required dependency and remove flax/jax from setup.

sayakpaul · sayakpaul · commit 522e70d3b3ab · 2025-08-26T15:10:23.000+02:00
diff --git a/setup.py b/setup.py
@@ -255,6 +255,7 @@ def run(self):
     deps["requests"],
     deps["safetensors"],
     deps["Pillow"],
+    deps["torch"],
 ]
 
 version_range_max = max(sys.version_info[1], 10) + 1

Original file line number	Diff line number	Diff line change
`@@ -255,6 +255,7 @@ def run(self):`
`255`	`255`	`deps["requests"],`
`256`	`256`	`deps["safetensors"],`
`257`	`257`	`deps["Pillow"],`
	`258`	`+ deps["torch"],`
`258`	`259`	`]`
`259`	`260`
`260`	`261`	`version_range_max = max(sys.version_info[1], 10) + 1`