Skip to content

Remove noop spmd_mode check, correct type annotations in attention flax - #170

Closed
coolkp wants to merge 2 commits into
mainfrom
code-health
Closed

Remove noop spmd_mode check, correct type annotations in attention flax#170
coolkp wants to merge 2 commits into
mainfrom
code-health

Fix type annotation mismatches in pil_utils.py and outputs.py

4005a8e
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs