File size: 288 Bytes
0558e79
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
from octo.model.octo_model import OctoModel
import os

# Set JAX to CPU (consistent with your setup)
os.environ['JAX_PLATFORMS'] = 'cpu'

# Load the model to cache it
model = OctoModel.load_pretrained("hf://rail-berkeley/octo-small-1.5")
print("Model downloaded and cached successfully.")