Model Card for Gia2

This is a multi-modal and multi-task model.

Model Details

Model Description

• Developed by: The GIA Team
• License: Apache 2.0

Model Sources

• Repository: https://github.com/huggingface/gia
• Paper: Coming soon
• Demo: Coming soon

Training

The model was trained on the following tasks:

• BabyAI-GoToRedBlueBall-v0
• BabyAI-Synth-v0
• BabyAI-Open-v0
• BabyAI-PickupAbove-v0
• BabyAI-OpenTwoDoors-v0
• BabyAI-BossLevel-v0

How to Get Started with the Model

Use the code below to get started with the model.

from transformers import AutoModelForCausalLM

model = AutoModelForCausalLM.from_pretrained("gia-project/gia2-small-babyai6")