Update README.md
Browse files
README.md
CHANGED
|
@@ -156,4 +156,6 @@ def quantize_model() -> None:
|
|
| 156 |
@app.local_entrypoint()
|
| 157 |
def main():
|
| 158 |
quantize_model.remote()
|
| 159 |
-
```
|
|
|
|
|
|
|
|
|
| 156 |
@app.local_entrypoint()
|
| 157 |
def main():
|
| 158 |
quantize_model.remote()
|
| 159 |
+
```
|
| 160 |
+
|
| 161 |
+
I think I also had to manually change something in the config.json -- needed to be "block_name_to_quantize": "model.transformer.blocks" instead of whatever it was, maybe "model_block_name_to_quantize" as the key or something.
|