[aisingapore#61] Added comment to run model_usage in local directory

ktyap · Nov 10, 2022 · eb76207 · eb76207
1 parent 421e862
commit eb76207
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sgnlp/models/dialogue_rnn/model_usage.py b/sgnlp/models/dialogue_rnn/model_usage.py
@@ -5,6 +5,10 @@
 from .postprocess import DialogueRNNPostprocessor
 
 
+"""To run in local directory ("models"):
+python -m dialogue_rnn.model_usage
+"""
+
 config = DialogueRNNConfig.from_pretrained("https://storage.googleapis.com/sgnlp/models/dialogue_rnn/config.json")
 model = DialogueRNNModel.from_pretrained("https://storage.googleapis.com/sgnlp/models/dialogue_rnn/pytorch_model.bin", config=config)