JetBrains
/

Mellum-4b-sft-python

Text Generation

text-generation-inference

Model card Files Files and versions

topshik commited on Apr 30

Commit

3b8d629

·

verified ·

1 Parent(s): 919a794

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -205,8 +205,8 @@ write_dist_matrix(dist_matrix, max_dist, output_data_filename_, verbose=True)
 end = time.perf_counter()
 """
-tokenizer = AutoTokenizer.from_pretrained('JetBrains/Mellum-4b-base')
-model = AutoModelForCausalLM.from_pretrained('JetBrains/Mellum-4b-base')
 encoded_input = tokenizer(example, return_tensors='pt', return_token_type_ids=False)
 input_len = len(encoded_input["input_ids"][0])
 out = model.generate(

 end = time.perf_counter()
 """
+tokenizer = AutoTokenizer.from_pretrained('JetBrains/Mellum-4b-sft-python')
+model = AutoModelForCausalLM.from_pretrained('JetBrains/Mellum-4b-sft-python')
 encoded_input = tokenizer(example, return_tensors='pt', return_token_type_ids=False)
 input_len = len(encoded_input["input_ids"][0])
 out = model.generate(