Update README.md
Browse files
README.md
CHANGED
@@ -11,35 +11,38 @@ base_model: tiiuae/falcon-7b
|
|
11 |
|
12 |
## Model Details
|
13 |
|
14 |
-
|
15 |
-
|
16 |
-
<!-- Provide a longer summary of what this model is. -->
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
- **Developed by:** [More Information Needed]
|
21 |
-
- **Funded by [optional]:** [More Information Needed]
|
22 |
-
- **Shared by [optional]:** [More Information Needed]
|
23 |
-
- **Model type:** [More Information Needed]
|
24 |
-
- **Language(s) (NLP):** [More Information Needed]
|
25 |
-
- **License:** [More Information Needed]
|
26 |
-
- **Finetuned from model [optional]:** [More Information Needed]
|
27 |
-
|
28 |
-
### Model Sources [optional]
|
29 |
-
|
30 |
-
<!-- Provide the basic links for the model. -->
|
31 |
-
|
32 |
-
- **Repository:** [More Information Needed]
|
33 |
-
- **Paper [optional]:** [More Information Needed]
|
34 |
-
- **Demo [optional]:** [More Information Needed]
|
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 |
|
40 |
### Direct Use
|
41 |
|
42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
43 |
|
44 |
[More Information Needed]
|
45 |
|
|
|
11 |
|
12 |
## Model Details
|
13 |
|
14 |
+
This model was finetuned on Business Manual Dataset
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
|
16 |
### Direct Use
|
17 |
|
18 |
+
```python
|
19 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
20 |
+
import transformers
|
21 |
+
import torch
|
22 |
+
|
23 |
+
model = "Dhruvil47/falcon-7b-bms"
|
24 |
+
|
25 |
+
tokenizer = AutoTokenizer.from_pretrained(model)
|
26 |
+
pipeline = transformers.pipeline(
|
27 |
+
"text-generation",
|
28 |
+
model=model,
|
29 |
+
tokenizer=tokenizer,
|
30 |
+
torch_dtype=torch.bfloat16,
|
31 |
+
trust_remote_code=True,
|
32 |
+
device_map="auto",
|
33 |
+
)
|
34 |
+
sequences = pipeline(
|
35 |
+
"Girafatron is obsessed with giraffes, the most glorious animal on the face of this Earth. Giraftron believes all other animals are irrelevant when compared to the glorious majesty of the giraffe.\nDaniel: Hello, Girafatron!\nGirafatron:",
|
36 |
+
max_length=200,
|
37 |
+
do_sample=True,
|
38 |
+
top_k=10,
|
39 |
+
num_return_sequences=1,
|
40 |
+
eos_token_id=tokenizer.eos_token_id,
|
41 |
+
)
|
42 |
+
for seq in sequences:
|
43 |
+
print(f"Result: {seq['generated_text']}")
|
44 |
+
|
45 |
+
```
|
46 |
|
47 |
[More Information Needed]
|
48 |
|