Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,11 @@
|
|
1 |
---
|
2 |
license: mit
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
4 |
|
5 |
## What is SmaliLLM used for
|
@@ -50,7 +56,7 @@ max_new_tokens=8192
|
|
50 |
)
|
51 |
output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
|
52 |
|
53 |
-
content = tokenizer.decode(output_ids
|
54 |
|
55 |
print("Java code:", content)
|
56 |
-
```
|
|
|
1 |
---
|
2 |
license: mit
|
3 |
+
base_model:
|
4 |
+
- Qwen/Qwen2.5-Coder-0.5B
|
5 |
+
pipeline_tag: text-generation
|
6 |
+
library_name: transformers
|
7 |
+
tags:
|
8 |
+
- code
|
9 |
---
|
10 |
|
11 |
## What is SmaliLLM used for
|
|
|
56 |
)
|
57 |
output_ids = generated_ids[0][len(model_inputs.input_ids[0]):].tolist()
|
58 |
|
59 |
+
content = tokenizer.decode(output_ids, skip_special_tokens=True).strip("\n")
|
60 |
|
61 |
print("Java code:", content)
|
62 |
+
```
|