Torch import was missing on line 120

README.md

@@ -117,6 +117,7 @@ You may also construct the pipeline from the loaded model and tokenizer yourself
 
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
 
 model_name = "h2oai/h2ogpt-gm-oasst1-en-2048-falcon-7b-v3"  # either local folder or huggingface model name
 # Important: The prompt needs to be in the same format the model was trained with.