Spaces:

ashioyajotham
/

falcon_7b_coder

Runtime error

ashioyajotham commited on Oct 5, 2023

Commit

dfbcf7f

•

1 Parent(s): 1f1c57f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,6 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
 fine_tuned_model = "ashioyajotham/results"
-model = AutoModelForCausalLM.from_pretrained(fine_tuned_model,trust_remote_code=True)
 # Load the tokenizer for the Falcon 7B model with remote code trust
 tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model, trust_remote_code=True)

+from transformers import AutoModelForCausalLM, AutoTokenizer, FalconForCausalLM
 fine_tuned_model = "ashioyajotham/results"
+model = FalconForCausalLM.from_pretrained(fine_tuned_model,trust_remote_code=True)
 # Load the tokenizer for the Falcon 7B model with remote code trust
 tokenizer = AutoTokenizer.from_pretrained(fine_tuned_model, trust_remote_code=True)