Update README.md
Browse files
README.md
CHANGED
@@ -62,7 +62,7 @@ fun main() {
|
|
62 |
"""
|
63 |
|
64 |
tokenizer = AutoTokenizer.from_pretrained('JetBrains/Mellum-4b-sft-kotlin')
|
65 |
-
model = AutoModelForCausalLM.from_pretrained('JetBrains/
|
66 |
encoded_input = tokenizer(example, return_tensors='pt', return_token_type_ids=False)
|
67 |
out = model.generate(
|
68 |
**encoded_input,
|
|
|
62 |
"""
|
63 |
|
64 |
tokenizer = AutoTokenizer.from_pretrained('JetBrains/Mellum-4b-sft-kotlin')
|
65 |
+
model = AutoModelForCausalLM.from_pretrained('JetBrains/Mellum-4b-sft-kotlin')
|
66 |
encoded_input = tokenizer(example, return_tensors='pt', return_token_type_ids=False)
|
67 |
out = model.generate(
|
68 |
**encoded_input,
|