Update README.md

README.md

@@ -13,7 +13,7 @@ widget:
     content: What is your favorite condiment?
 license: other
 datasets:
-- XuehangCang/Erotic-literature-split
+- XuehangCang/Erotic-literature
 language:
 - zh
 ---
@@ -22,14 +22,13 @@ language:
 
 ## Overview
 
-This model is a fine-tuned version of the `meta-llama/Llama-3.1-8B` using the PEFT (Parameter Efficient Fine-Tuning) approach on a specialized dataset for generating text in the genre of erotic literature. It has been designed to understand and generate text in Chinese related to this specific domain, making it a useful tool for tasks involving creative writing, story generation, or other applications within this genre.
-
+This model is a fine-tuned version of the `meta-llama/Llama-3.1-8B`, using the SFT approach on the [XuehangCang/Erotic-literature](https://huggingface.co/datasets/XuehangCang/Erotic-literature) dataset. It's specifically crafted to understand and generate text in Chinese related to the erotic literature genre. This makes it an invaluable tool for tasks involving creative writing, story generation, or other applications within this specific domain.
 ## Model Details
 
 - **Base Model**: `meta-llama/Llama-3.1-8B`
 - **Model Type**: Text Generation
 - **Language**: Chinese (zh)
-- **Fine-Tuning**: The model has been fine-tuned using the `XuehangCang/Erotic-literature-split` dataset which specifically focuses on erotic literature. This allows the model to generate relevant and cohesive content within this genre.
+- **Fine-Tuning**: The model has been fine-tuned using the `XuehangCang/Erotic-literature` dataset which specifically focuses on erotic literature. This allows the model to generate relevant and cohesive content within this genre.
 
 ## Usage
 
@@ -38,16 +37,21 @@ This model is a fine-tuned version of the `meta-llama/Llama-3.1-8B` using the PE
 This model can be utilized through the Transformers library. Below is a basic example of how to use this model for text generation:
 
 ```python
-from transformers import pipeline
+import transformers
+import torch
+
+model_id = "XuehangCang/autotrain-fgk68-62aj5"
+
+pipeline = transformers.pipeline(
+    "text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto"
+)
+
+print(pipeline("在星光熠熠的夜晚，她独自在海边徘徊，思绪纷飞。", max_length=150))
 
-# Load the model
-generator = pipeline('text-generation', model='your_username/Erotic-literature-model')
+"""
 
-# Generate text
-prompt = "在星光熠熠的夜晚，她独自在海边徘徊，思绪纷飞。"
-output = generator(prompt, max_length=100, num_return_sequences=1)
+"""
 
-print(output[0]['generated_text'])
 ```
 
 ### Applications