From f4c96b7339c5efffd1d5f3b08cf947d9e7787a4b Mon Sep 17 00:00:00 2001
From: Michael Cusack <michael.cusack@color.com>
Date: Fri, 4 Aug 2023 23:11:33 +0700
Subject: [PATCH] Add options to save_torchscript

---
 save_torchscript.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/save_torchscript.py b/save_torchscript.py
index cb520d8..5b7f1ed 100755
--- a/save_torchscript.py
+++ b/save_torchscript.py
@@ -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())