update fiddle requirements

Author: aireenmei

praxis/pip_package/cloudbuild-release.yaml

@@ -22,6 +22,7 @@ substitutions:
 options:
     dynamic_substitutions: true
     substitution_option: 'ALLOW_LOOSE'
+    machineType: E2_HIGHCPU_8
 timeout: 5400s
 artifacts:
   objects:

praxis/pip_package/requirements.in

@@ -4,7 +4,7 @@
 absl-py
 clu
 einops
-fiddle @ git+https://github.com/google/fiddle
+fiddle
 flax
 jax
 jax-bitempered-loss
@@ -15,4 +15,4 @@ optax-shampoo
 protobuf==3.15
 t5x
 tensorflow~=2.9.2
-tensorflow-text~=2.9.0
+tensorflow-text~=2.9.0

praxis/pip_package/requirements.txt

@@ -83,7 +83,7 @@ executing==1.2.0
     # via stack-data
 fastjsonschema==2.16.2
     # via nbformat
-fiddle @ git+https://github.com/google/fiddle
+fiddle==0.2.4
     # via -r praxis-requirements.in
 flatbuffers==1.12
     # via tensorflow