davidberenstein1957 commited on
Commit
93a1fa9
·
verified ·
1 Parent(s): 1907e26

Add files using upload-large-folder tool

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. smash_config.json +1 -1
README.md CHANGED
@@ -38,7 +38,7 @@ The compression configuration of the model is stored in the `smash_config.json`
38
  "pruner": null,
39
  "quantizer": null,
40
  "max_batch_size": 1,
41
- "device": "cuda",
42
  "save_fns": [],
43
  "load_fns": [
44
  "transformers"
 
38
  "pruner": null,
39
  "quantizer": null,
40
  "max_batch_size": 1,
41
+ "device": "cpu",
42
  "save_fns": [],
43
  "load_fns": [
44
  "transformers"
smash_config.json CHANGED
@@ -5,7 +5,7 @@
5
  "pruner": null,
6
  "quantizer": null,
7
  "max_batch_size": 1,
8
- "device": "cuda",
9
  "save_fns": [],
10
  "load_fns": [
11
  "transformers"
 
5
  "pruner": null,
6
  "quantizer": null,
7
  "max_batch_size": 1,
8
+ "device": "cpu",
9
  "save_fns": [],
10
  "load_fns": [
11
  "transformers"