Commit
•
da8f261
1
Parent(s):
980a6e5
Update README.md
Browse files
README.md
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
---
|
| 2 |
language:
|
| 3 |
- en
|
|
|
|
| 4 |
---
|
| 5 |
This is llama2 7B finetuned using qlora with bf16 as compute dtype. The dataset has been generated using open-ai api with samples semantics oriented towards abstract explanation of system design.
|
| 6 |
|
|
@@ -23,5 +24,4 @@ prompt = "Design an application like Whatsapp with tech stack you will use"
|
|
| 23 |
gen = pipeline('text-generation', model=model, tokenizer=tokenizer)
|
| 24 |
result = gen(prompt)
|
| 25 |
print(result[0]['generated_text'])
|
| 26 |
-
```
|
| 27 |
-
|
|
|
|
| 1 |
---
|
| 2 |
language:
|
| 3 |
- en
|
| 4 |
+
pipeline_tag: text-generation
|
| 5 |
---
|
| 6 |
This is llama2 7B finetuned using qlora with bf16 as compute dtype. The dataset has been generated using open-ai api with samples semantics oriented towards abstract explanation of system design.
|
| 7 |
|
|
|
|
| 24 |
gen = pipeline('text-generation', model=model, tokenizer=tokenizer)
|
| 25 |
result = gen(prompt)
|
| 26 |
print(result[0]['generated_text'])
|
| 27 |
+
```
|
|
|