Add options to save_torchscript

2023-08-04 23:11:33 +07:00
parent 4b3a41b8fc
commit f4c96b7339
1 changed files with 2 additions and 2 deletions
@@ -25,7 +25,7 @@ import torch
 from model import ModelArgs, Transformer
-# Model args
+# Model args config
 dim = 288
 n_layers = 6
 n_heads = 6
@@ -39,7 +39,7 @@ norm_eps = 1e-5
 model_path = "model.pt"
 zero_params = False
 gzip_output = False
-
+# Allow config overrides
 exec(open("configurator.py").read())