How to use rail-berkeley/octo-small with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("rail-berkeley/octo-small", device_map="auto")
