doberst commited on
Commit
5d16507
1 Parent(s): 4a8dda5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -72,8 +72,8 @@ Any model can provide inaccurate or incomplete information, and should be used i
72
  The fastest way to get started with dRAGon is through direct import in transformers:
73
 
74
  from transformers import AutoTokenizer, AutoModelForCausalLM
75
- tokenizer = AutoTokenizer.from_pretrained("dragon-deci-7b-v0")
76
- model = AutoModelForCausalLM.from_pretrained("dragon-deci-7b-v0")
77
 
78
  Please refer to the generation_test .py files in the Files repository, which includes 200 samples and script to test the model. The **generation_test_llmware_script.py** includes built-in llmware capabilities for fact-checking, as well as easy integration with document parsing and actual retrieval to swap out the test set for RAG workflow consisting of business documents.
79
 
 
72
  The fastest way to get started with dRAGon is through direct import in transformers:
73
 
74
  from transformers import AutoTokenizer, AutoModelForCausalLM
75
+ tokenizer = AutoTokenizer.from_pretrained("dragon-deci-7b-v0", trust_remote_code=True)
76
+ model = AutoModelForCausalLM.from_pretrained("dragon-deci-7b-v0", trust_remote_code=True)
77
 
78
  Please refer to the generation_test .py files in the Files repository, which includes 200 samples and script to test the model. The **generation_test_llmware_script.py** includes built-in llmware capabilities for fact-checking, as well as easy integration with document parsing and actual retrieval to swap out the test set for RAG workflow consisting of business documents.
79