Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,7 @@
|
|
1 |
---
|
2 |
-
base_model:
|
|
|
|
|
3 |
tags:
|
4 |
- text-generation-inference
|
5 |
- transformers
|
@@ -87,5 +89,4 @@ tokenizer = AutoTokenizer.from_pretrained(base_model)
|
|
87 |
model = AutoModelForCausalLM.from_pretrained(base_model)
|
88 |
model = PeftModel.from_pretrained(model, lora_adapter)
|
89 |
|
90 |
-
model.eval()
|
91 |
-
|
|
|
1 |
---
|
2 |
+
base_model:
|
3 |
+
- unsloth/phi-4-unsloth-bnb-4bit
|
4 |
+
- microsoft/phi-4
|
5 |
tags:
|
6 |
- text-generation-inference
|
7 |
- transformers
|
|
|
89 |
model = AutoModelForCausalLM.from_pretrained(base_model)
|
90 |
model = PeftModel.from_pretrained(model, lora_adapter)
|
91 |
|
92 |
+
model.eval()
|
|