Update README.md
Browse files
README.md
CHANGED
|
@@ -1,6 +1,9 @@
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
pipeline_tag: text-generation
|
|
|
|
|
|
|
|
|
|
| 4 |
---
|
| 5 |
|
| 6 |
|
|
@@ -29,7 +32,7 @@ We release SongComposer series in two versions:
|
|
| 29 |
To load the SongComposer_sft model using Transformers, use the following code:
|
| 30 |
```python
|
| 31 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 32 |
-
ckpt_path = "Mar2Ding/
|
| 33 |
tokenizer = AutoTokenizer.from_pretrained(ckpt_path, trust_remote_code=True)
|
| 34 |
model = AutoModel.from_pretrained(ckpt_path, trust_remote_code=True).cuda().half()
|
| 35 |
prompt = 'Create a song on brave and sacrificing with a rapid pace.'
|
|
@@ -41,7 +44,7 @@ model.inference(prompt, tokenizer)
|
|
| 41 |
|
| 42 |
```python
|
| 43 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 44 |
-
ckpt_path = "Mar2Ding/
|
| 45 |
tokenizer = AutoTokenizer.from_pretrained(ckpt_path, trust_remote_code=True)
|
| 46 |
model = AutoModel.from_pretrained(ckpt_path, trust_remote_code=True).cuda().half()
|
| 47 |
prompt = 'Create a song on brave and sacrificing with a rapid pace.'
|
|
|
|
| 1 |
---
|
| 2 |
license: apache-2.0
|
| 3 |
pipeline_tag: text-generation
|
| 4 |
+
language:
|
| 5 |
+
- en
|
| 6 |
+
- zh
|
| 7 |
---
|
| 8 |
|
| 9 |
|
|
|
|
| 32 |
To load the SongComposer_sft model using Transformers, use the following code:
|
| 33 |
```python
|
| 34 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 35 |
+
ckpt_path = "Mar2Ding/songcomposer_sft"
|
| 36 |
tokenizer = AutoTokenizer.from_pretrained(ckpt_path, trust_remote_code=True)
|
| 37 |
model = AutoModel.from_pretrained(ckpt_path, trust_remote_code=True).cuda().half()
|
| 38 |
prompt = 'Create a song on brave and sacrificing with a rapid pace.'
|
|
|
|
| 44 |
|
| 45 |
```python
|
| 46 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
| 47 |
+
ckpt_path = "Mar2Ding/songcomposer_sft"
|
| 48 |
tokenizer = AutoTokenizer.from_pretrained(ckpt_path, trust_remote_code=True)
|
| 49 |
model = AutoModel.from_pretrained(ckpt_path, trust_remote_code=True).cuda().half()
|
| 50 |
prompt = 'Create a song on brave and sacrificing with a rapid pace.'
|