example code missing `import torch` (#1)

- example code missing `import torch` (2bfa9e54adf1ba3c37021615ff9cdedb44b53228)


Co-authored-by: Yassine Ennaour <[email protected]>

README.md

@@ -17,6 +17,7 @@ Here give some examples of how to use our model:
 
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
 
 prompt = '''def min(arr):
     """