sohamslc5 commited on
Commit
5a097b1
1 Parent(s): a6cc8ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,4 +1,6 @@
1
  # Load model directly
 
 
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
  tokenizer = AutoTokenizer.from_pretrained("sohamslc5/llama_2_7B_sohamslc5")
 
1
  # Load model directly
2
+ !pip install transformers==4.30.0
3
+
4
  from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
  tokenizer = AutoTokenizer.from_pretrained("sohamslc5/llama_2_7B_sohamslc5")