Spaces:
Running
Running
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.") |