Update README.md
Browse files
README.md
CHANGED
@@ -64,13 +64,8 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
64 |
# Load tokenizer
|
65 |
tokenizer = AutoTokenizer.from_pretrained(adapter_path)
|
66 |
```
|
67 |
-
|
68 |
-
|
69 |
-
input_text = "Your input text here"
|
70 |
-
inputs = tokenizer(input_text, return_tensors="pt").to(DEV)
|
71 |
-
outputs = model.generate(**inputs)
|
72 |
-
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
73 |
-
```
|
74 |
|
75 |
## Training Details
|
76 |
|
|
|
64 |
# Load tokenizer
|
65 |
tokenizer = AutoTokenizer.from_pretrained(adapter_path)
|
66 |
```
|
67 |
+
|
68 |
+
Refer to the following repository for further usage of the model : https://github.com/mzbac/llama2-fine-tune
|
|
|
|
|
|
|
|
|
|
|
69 |
|
70 |
## Training Details
|
71 |
|