Update README.md
Browse files
README.md
CHANGED
@@ -5,7 +5,8 @@ base_model: mistralai/Mistral-7B-v0.1
|
|
5 |
|
6 |
# Model Card for Model ID
|
7 |
|
8 |
-
|
|
|
9 |
|
10 |
|
11 |
|
@@ -17,13 +18,12 @@ base_model: mistralai/Mistral-7B-v0.1
|
|
17 |
|
18 |
|
19 |
|
20 |
-
- **Developed by:**
|
21 |
-
- **Funded by [optional]:**
|
22 |
-
- **
|
23 |
-
- **
|
24 |
-
- **
|
25 |
-
- **
|
26 |
-
- **Finetuned from model [optional]:** [More Information Needed]
|
27 |
|
28 |
### Model Sources [optional]
|
29 |
|
@@ -35,43 +35,59 @@ base_model: mistralai/Mistral-7B-v0.1
|
|
35 |
|
36 |
## Uses
|
37 |
|
38 |
-
|
39 |
-
|
40 |
### Direct Use
|
41 |
|
42 |
-
|
43 |
-
|
44 |
-
[More Information Needed]
|
45 |
|
46 |
### Downstream Use [optional]
|
47 |
|
48 |
-
|
49 |
-
|
50 |
-
[More Information Needed]
|
51 |
|
52 |
### Out-of-Scope Use
|
53 |
|
54 |
-
|
55 |
-
|
56 |
-
[More Information Needed]
|
57 |
|
58 |
## Bias, Risks, and Limitations
|
59 |
|
60 |
-
|
61 |
-
|
62 |
-
[More Information Needed]
|
63 |
|
64 |
### Recommendations
|
65 |
|
66 |
-
|
67 |
-
|
68 |
-
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
69 |
|
70 |
## How to Get Started with the Model
|
71 |
|
72 |
Use the code below to get started with the model.
|
73 |
|
74 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
75 |
|
76 |
## Training Details
|
77 |
|
|
|
5 |
|
6 |
# Model Card for Model ID
|
7 |
|
8 |
+
Our finetuned Mistral LLM is a large language model specialized for natural language processing tasks, delivering enhanced performance for a
|
9 |
+
wide array of applications, including text classification, question-answering, chatbot services, and more.
|
10 |
|
11 |
|
12 |
|
|
|
18 |
|
19 |
|
20 |
|
21 |
+
- **Developed by:** Basel Anaya, Osama Awad, Yazeed Mshayekh
|
22 |
+
- **Funded by [optional]:** Basel Anaya, Osama Awad, Yazeed Mshayekh
|
23 |
+
- **Model type:** Autoregressive Language Model
|
24 |
+
- **Language(s) (NLP):** English
|
25 |
+
- **License:** MIT License
|
26 |
+
- **Finetuned from model:** MistralAI's Mistral-7B
|
|
|
27 |
|
28 |
### Model Sources [optional]
|
29 |
|
|
|
35 |
|
36 |
## Uses
|
37 |
|
38 |
+
Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model.
|
|
|
39 |
### Direct Use
|
40 |
|
41 |
+
Users can leverage the finetuned Mistral LLM for various NLP tasks right out-of-the-box. Simply interact with the API or load the model locally to experience superior language understanding and generation capabilities. Ideal for developers seeking rapid prototyping and deployment of conversational AI applications.
|
|
|
|
|
42 |
|
43 |
### Downstream Use [optional]
|
44 |
|
45 |
+
Integrate the finetuned Mistral LLM effortlessly into custom applications and pipelines. Utilize the model as a starting point for further refinement, targeting industry-specific lingo, niches, or particular use cases. Seamless compatibility ensures smooth collaboration with adjacent technologies and services.
|
|
|
|
|
46 |
|
47 |
### Out-of-Scope Use
|
48 |
|
49 |
+
Limitations exist concerning controversial topics, sensitive data, and scenarios demanding real-time responses. Users should exercise caution when deploying the model in safety-critical situations or regions with strict compliance regulations. Avoid sharing confidential or personally identifiable information with the model.
|
|
|
|
|
50 |
|
51 |
## Bias, Risks, and Limitations
|
52 |
|
53 |
+
Address both technical and sociotechnical limitations.
|
|
|
|
|
54 |
|
55 |
### Recommendations
|
56 |
|
57 |
+
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. Further recommendations include cautious assessment of ethical implications, ongoing maintenance, periodic evaluations, and responsible reporting practices.
|
|
|
|
|
58 |
|
59 |
## How to Get Started with the Model
|
60 |
|
61 |
Use the code below to get started with the model.
|
62 |
|
63 |
+
```python
|
64 |
+
import torch
|
65 |
+
from transformers import pipeline, AutoTokenizer
|
66 |
+
|
67 |
+
# Load the finetuned Mistral LLM
|
68 |
+
model_name = "Reverb/Mistral-7B-LoreWeaver"
|
69 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
70 |
+
generator = pipeline("text-generation", model=model_name, tokenizer=tokenizer)
|
71 |
+
|
72 |
+
# Example usage
|
73 |
+
input_text = "Once upon a time,"
|
74 |
+
num_generated_tokens = 50
|
75 |
+
|
76 |
+
response = generator(input_text, max_length=num_generated_tokens, num_return_sequences=1)
|
77 |
+
print(f"Generated text:\n{response[0]['generated_text']}")
|
78 |
+
|
79 |
+
# Alternatively, for fine-grained control over the generation process
|
80 |
+
inputs = tokenizer(input_text, return_tensors="pt")
|
81 |
+
outputs = generator.generate(
|
82 |
+
inputs["input_ids"].to("cuda"),
|
83 |
+
max_length=num_generated_tokens,
|
84 |
+
num_beams=5,
|
85 |
+
early_stopping=True,
|
86 |
+
temperature=1.2,
|
87 |
+
)
|
88 |
+
generated_sentence = tokenizer.decode(outputs[0])
|
89 |
+
print(f"\nGenerated text with beam search and custom params:\n{generated_sentence}")
|
90 |
+
```
|
91 |
|
92 |
## Training Details
|
93 |
|