produc-xuan commited on
Commit
337600e
·
verified ·
1 Parent(s): a05b1c2

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. config.json +1 -1
  2. model.safetensors +1 -1
  3. train_config.json +1 -1
config.json CHANGED
@@ -38,7 +38,7 @@
38
  ]
39
  }
40
  },
41
- "device": "mps",
42
  "use_amp": false,
43
  "chunk_size": 100,
44
  "n_action_steps": 100,
 
38
  ]
39
  }
40
  },
41
+ "device": "cuda",
42
  "use_amp": false,
43
  "chunk_size": 100,
44
  "n_action_steps": 100,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2d3619bab2372d2c142fa6d31e5fc83d5c02d97fde17ae3d845e4f4934ce071a
3
  size 206701072
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:14954a89efe0e08903e0fa8da11faeef566103c2521345aad1bb9210a20652bd
3
  size 206701072
train_config.json CHANGED
@@ -105,7 +105,7 @@
105
  ]
106
  }
107
  },
108
- "device": "mps",
109
  "use_amp": false,
110
  "chunk_size": 100,
111
  "n_action_steps": 100,
 
105
  ]
106
  }
107
  },
108
+ "device": "cuda",
109
  "use_amp": false,
110
  "chunk_size": 100,
111
  "n_action_steps": 100,