Update README.md
Browse files
README.md
CHANGED
|
@@ -5,6 +5,8 @@ license: apache-2.0
|
|
| 5 |
base_model: mistralai/Mistral-7B-Instruct-v0.3
|
| 6 |
library_name: peft
|
| 7 |
pipeline_tag: text-generation
|
|
|
|
|
|
|
| 8 |
tags:
|
| 9 |
- lora
|
| 10 |
- llamafactory
|
|
@@ -53,7 +55,7 @@ model = PeftModel.from_pretrained(base_model, lora_model_name)
|
|
| 53 |
model.eval()
|
| 54 |
|
| 55 |
# 3. Generate Text
|
| 56 |
-
input_text = "
|
| 57 |
inputs = tokenizer(input_text, return_tensors="pt").to("cuda")
|
| 58 |
|
| 59 |
with torch.no_grad():
|
|
|
|
| 5 |
base_model: mistralai/Mistral-7B-Instruct-v0.3
|
| 6 |
library_name: peft
|
| 7 |
pipeline_tag: text-generation
|
| 8 |
+
datasets:
|
| 9 |
+
- MattCoddity/dockerNLcommands
|
| 10 |
tags:
|
| 11 |
- lora
|
| 12 |
- llamafactory
|
|
|
|
| 55 |
model.eval()
|
| 56 |
|
| 57 |
# 3. Generate Text
|
| 58 |
+
input_text = "I want to see the running containers with the ACME label from the vendor ACME."
|
| 59 |
inputs = tokenizer(input_text, return_tensors="pt").to("cuda")
|
| 60 |
|
| 61 |
with torch.no_grad():
|