test.py

ORGANIZATION = 'google'

BASE_MODELS = [
    ('paligemma-3b-mix-224-jax', 'paligemma-3b-mix-224'),
    ('paligemma-3b-mix-448-jax', 'paligemma-3b-mix-448'),
]
MODELS = {
    **{
        model_name: (
            f'{ORGANIZATION}/{repo}',
            f'{model_name}.bf16.npz',
            'bfloat16',  # Model repo revision.
        )
        for repo, model_name in BASE_MODELS
    },
    'testPaligemma':('Jegree/myPaligem', 'fine-tuned-paligemma-3b-pt-224.f16.npz'),
}

print(MODELS)