Hugging Face's logo

Spaces:
Nirav-Madhani
/
Octo-1.5-Base
Sleeping

App Files Community
Nirav-Madhani commited on
Commit
bae578d
·
verified ·
1 Parent(s): 8a209a5

Update init_model.py

Browse files
Files changed (1) hide show
  1. init_model.py +1 -1
init_model.py CHANGED
@@ -5,5 +5,5 @@ import os
5
  os.environ['JAX_PLATFORMS'] = 'cpu'
6
 
7
  # Load the model to cache it
8
- model = OctoModel.load_pretrained("hf://rail-berkeley/octo-small-1.5")
9
  print("Model downloaded and cached successfully.")
 
5
  os.environ['JAX_PLATFORMS'] = 'cpu'
6
 
7
  # Load the model to cache it
8
+ model = OctoModel.load_pretrained("hf://rail-berkeley/octo-base-1.5")
9
  print("Model downloaded and cached successfully.")