TechxGenus Lyte commited on
Commit
d779188
1 Parent(s): 690f8ff

example code missing `import torch` (#1)

Browse files

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


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

Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -17,6 +17,7 @@ Here give some examples of how to use our model:
17
 
18
  ```python
19
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
20
 
21
  prompt = '''def min(arr):
22
  """
 
17
 
18
  ```python
19
  from transformers import AutoTokenizer, AutoModelForCausalLM
20
+ import torch
21
 
22
  prompt = '''def min(arr):
23
  """