I am getting a different error now:
/home/ec2-user/anaconda3/envs/JupyterSystemEnv/lib/python3.7/site-packages/flax/core/lift.py:112: FutureWarning: jax.tree_flatten is deprecated, and will be removed in a future release. Use jax.tree_util.tree_flatten instead. scopes, treedef = jax.tree_flatten(scope_tree) /home/ec2-user/anaconda3/envs/JupyterSystemEnv/lib/python3.7/site-packages/flax/core/lift.py:729: FutureWarning: jax.tree_leaves is deprecated, and will be removed in a future release. Use jax.tree_util.tree_leaves instead. lengths = set(jax.tree_leaves(lengths)) /home/ec2-user/anaconda3/envs/JupyterSystemEnv/lib/python3.7/site-packages/flax/core/axes_scan.py:134: FutureWarning: jax.tree_flatten is deprecated, and will be removed in a future release. Use jax.tree_util.tree_flatten instead. in_avals, in_tree = jax.tree_flatten(input_avals) /home/ec2-user/anaconda3/envs/JupyterSystemEnv/lib/python3.7/site-packages/flax/linen/transforms.py:249: FutureWarning: jax.tree_leaves is deprecated, and will be removed in a future release. Use jax.tree_util.tree_leaves instead. jax.tree_leaves(tree))) /home/ec2-user/anaconda3/envs/JupyterSystemEnv/lib/python3.7/site-packages/flax/core/axes_scan.py:146: FutureWarning: jax.tree_unflatten is deprecated, and will be removed in a future release. Use jax.tree_util.tree_unflatten instead. broadcast_in, constants_out = jax.tree_unflatten(out_tree(), out_flat) Killed
I have imported the library in the inference.py script:
from jax.tree_util import (tree_flatten, tree_leaves, tree_unflatten)
But still I am experiencing this warning and the running of code is interrupts and the session gets killed.
Any reason. Please let me know.
How do I access CloudFormation stack name from the Macro Handler?asked 18 days ago
ImportError: cannot import name 'dataclass_transform' from 'typing_extensions' (/home/ec2-user/anaconda3/envs/tensorflow2_p38/lib/python3.8/site-packages/typing_extensions.py)asked 22 days ago
MWAA apache-airflow-providers-amazon DAGS can't import operatorasked 2 months ago
Is there a programmatic way to download the EC2 Instance Recommendations from Migration Hub?asked 4 months ago
AWS Glue ETL Job: IllegalArgumentException: Missing collection name.asked 3 months ago
Cannot import numba python package inside AWS Lambda functionasked 4 months ago
Unable to Import Custom slot type from a zip fileasked 6 months ago
some of the command from bash script is not runningasked 2 months ago
How can I delete dataset import job?asked 2 years ago
Import Contact Flow via CLI or UIasked 2 years ago