Update README.md
Browse files
README.md
CHANGED
|
@@ -52,7 +52,7 @@ Each training row was converted into:
|
|
| 52 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 53 |
import torch
|
| 54 |
|
| 55 |
-
model_id = "Shlok307/
|
| 56 |
|
| 57 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 58 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
|
| 52 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 53 |
import torch
|
| 54 |
|
| 55 |
+
model_id = "Shlok307/ai_interview-lora"
|
| 56 |
|
| 57 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
| 58 |
model = AutoModelForCausalLM.from_pretrained(
|