Update README.md

README.md

@@ -30,6 +30,41 @@ Demo: [🤗 Space with ZERO](https://huggingface.co/spaces/p1atdev/danbooru-tags
 
 ## Usage
 
+## Using 🤗Transformers
+
+```py
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+MODEL_NAME = "p1atdev/dart-v2-moe-sft"
+
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.bfloat16)
+
+prompt = (
+    f"<|bos|>"
+    f"<copyright>vocaloid</copyright>"
+    f"<character>hatsune miku</character>"
+    f"<|rating:general|><|aspect_ratio:tall|><|length:long|>"
+    f"<general>1girl, cat ears<|identity:none|><|input_end|>"
+)
+inputs = tokenizer(prompt, return_tensors="pt").input_ids
+
+with torch.no_grad():
+  outputs = model.generate(
+    inputs,
+    do_sample=True,
+    temperature=1.0,
+    top_p=1.0,
+    top_k=100,
+    max_new_tokens=128,
+    num_beams=1,
+  )
+
+print(", ".join([tag for tag in tokenizer.batch_decode(outputs[0], skip_special_tokens=True) if tag.strip() != ""]))
+# vocaloid, hatsune miku, 1girl, cat ears, closed mouth, detached sleeves, dress, expressionless, from behind, full body, green theme, hair ornament, hair ribbon, headphones, high heels, holding, holding microphone, long hair, microphone, monochrome, necktie, ribbon, short dress, shoulder tattoo, simple background, sleeveless, sleeveless dress, spot color, standing, tattoo, thighhighs, twintails, very long hair, white background
+```
+
 ### Using 📦`dartrs` library
 
 > [!WARNING]
@@ -80,41 +115,6 @@ print(f"Time taken: {end - start:.2f}s")
 # Time taken: 0.26s
 ```
 
-## Using 🤗Transformers
-
-```py
-import torch
-from transformers import AutoTokenizer, AutoModelForCausalLM
-
-MODEL_NAME = "p1atdev/dart-v2-moe-sft"
-
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.bfloat16)
-
-prompt = (
-    f"<|bos|>"
-    f"<copyright>vocaloid</copyright>"
-    f"<character>hatsune miku</character>"
-    f"<|rating:general|><|aspect_ratio:tall|><|length:long|>"
-    f"<general>1girl, cat ears<|identity:none|><|input_end|>"
-)
-inputs = tokenizer(prompt, return_tensors="pt").input_ids
-
-with torch.no_grad():
-  outputs = model.generate(
-    inputs,
-    do_sample=True,
-    temperature=1.0,
-    top_p=1.0,
-    top_k=100,
-    max_new_tokens=128,
-    num_beams=1,
-  )
-
-print(", ".join([tag for tag in tokenizer.batch_decode(outputs[0], skip_special_tokens=True) if tag.strip() != ""]))
-# vocaloid, hatsune miku, 1girl, cat ears, closed mouth, detached sleeves, dress, expressionless, from behind, full body, green theme, hair ornament, hair ribbon, headphones, high heels, holding, holding microphone, long hair, microphone, monochrome, necktie, ribbon, short dress, shoulder tattoo, simple background, sleeveless, sleeveless dress, spot color, standing, tattoo, thighhighs, twintails, very long hair, white background
-```
-
 ## Prompt Format
 
 ```py