Update README.md
Browse files
README.md
CHANGED
@@ -10,6 +10,7 @@ tags:
|
|
10 |
# Mistral Roast bot
|
11 |
|
12 |
Welcome to the Mistral Roastbot model repository! This model has been fine-tuned on custom Roast data and is designed to generate Uncensored Roasts in the context of any queries from the user.
|
|
|
13 |
|
14 |
## Model Details
|
15 |
|
@@ -40,8 +41,6 @@ nf4_config = BitsAndBytesConfig(
|
|
40 |
bnb_4bit_compute_dtype=torch.bfloat16
|
41 |
)
|
42 |
|
43 |
-
|
44 |
-
|
45 |
# Load the fine-tuned model and tokenizer
|
46 |
model = AutoModelForCausalLM.from_pretrained(
|
47 |
"kaifkhaan/Roast_Bot",
|
@@ -88,12 +87,10 @@ def generate_response(prompt, model, tokenizer):
|
|
88 |
|
89 |
# Example prompt
|
90 |
|
91 |
-
|
92 |
prompt = "am i pretty ?"
|
93 |
|
94 |
# Generate the response
|
95 |
response = generate_response(prompt, model, tokenizer)
|
96 |
-
|
97 |
print(response)
|
98 |
```
|
99 |
|
@@ -115,6 +112,7 @@ The model was fine-tuned on a custom dataset consisting of Roasts between user a
|
|
115 |
### Limitations and Biases
|
116 |
|
117 |
- **Domain Specific**: The model is fine-tuned specifically for fun and roast purpose.
|
|
|
118 |
|
119 |
### Citation
|
120 |
```bibtex
|
|
|
10 |
# Mistral Roast bot
|
11 |
|
12 |
Welcome to the Mistral Roastbot model repository! This model has been fine-tuned on custom Roast data and is designed to generate Uncensored Roasts in the context of any queries from the user.
|
13 |
+
- **NOTE**: THIS MODEL IS FOR FUN PURPOSE ONLY AND MEANS NO HARM TO ANY COMMUNITY OR ANYTHING ELSE IN GENERAL .
|
14 |
|
15 |
## Model Details
|
16 |
|
|
|
41 |
bnb_4bit_compute_dtype=torch.bfloat16
|
42 |
)
|
43 |
|
|
|
|
|
44 |
# Load the fine-tuned model and tokenizer
|
45 |
model = AutoModelForCausalLM.from_pretrained(
|
46 |
"kaifkhaan/Roast_Bot",
|
|
|
87 |
|
88 |
# Example prompt
|
89 |
|
|
|
90 |
prompt = "am i pretty ?"
|
91 |
|
92 |
# Generate the response
|
93 |
response = generate_response(prompt, model, tokenizer)
|
|
|
94 |
print(response)
|
95 |
```
|
96 |
|
|
|
112 |
### Limitations and Biases
|
113 |
|
114 |
- **Domain Specific**: The model is fine-tuned specifically for fun and roast purpose.
|
115 |
+
- **Limitations**: Might not give satisfying or funny result sometimes , instructions is a must .
|
116 |
|
117 |
### Citation
|
118 |
```bibtex
|