How to use iamthe66epitaph/BabyAI with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="iamthe66epitaph/BabyAI")
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("iamthe66epitaph/BabyAI", dtype="auto")
8193465
1
2
3
4
5
6
7
8
9
#!/home/joker/babyai/bin/python3 # -*- coding: utf-8 -*- import re import sys from transformers.commands.transformers_cli import main_cli if __name__ == '__main__': sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0]) sys.exit(main_cli())
