Views
No views yet
| Name | Quant method | Size |
|---|---|---|
| Chat_GPT-2.Q2_K.gguf | Q2_K | 0.08GB |
| Chat_GPT-2.IQ3_XS.gguf | IQ3_XS | 0.08GB |
| Chat_GPT-2.IQ3_S.gguf | IQ3_S | 0.08GB |
| Chat_GPT-2.Q3_K_S.gguf | Q3_K_S | 0.08GB |
| Chat_GPT-2.IQ3_M.gguf | IQ3_M | 0.09GB |
| Chat_GPT-2.Q3_K.gguf | Q3_K | 0.09GB |
| Chat_GPT-2.Q3_K_M.gguf | Q3_K_M | 0.09GB |
| Chat_GPT-2.Q3_K_L.gguf | Q3_K_L | 0.1GB |
| Chat_GPT-2.IQ4_XS.gguf | IQ4_XS | 0.1GB |
| Chat_GPT-2.Q4_0.gguf | Q4_0 | 0.1GB |
| Chat_GPT-2.IQ4_NL.gguf | IQ4_NL | 0.1GB |
| Chat_GPT-2.Q4_K_S.gguf | Q4_K_S | 0.1GB |
| Chat_GPT-2.Q4_K.gguf | Q4_K | 0.11GB |
| Chat_GPT-2.Q4_K_M.gguf | Q4_K_M | 0.11GB |
| Chat_GPT-2.Q4_1.gguf | Q4_1 | 0.11GB |
| Chat_GPT-2.Q5_0.gguf | Q5_0 | 0.11GB |
| Chat_GPT-2.Q5_K_S.gguf | Q5_K_S | 0.11GB |
| Chat_GPT-2.Q5_K.gguf | Q5_K | 0.12GB |
| Chat_GPT-2.Q5_K_M.gguf | Q5_K_M | 0.12GB |
| Chat_GPT-2.Q5_1.gguf | Q5_1 | 0.12GB |
| Chat_GPT-2.Q6_K.gguf | Q6_K | 0.13GB |
| Chat_GPT-2.Q8_0.gguf | Q8_0 | 0.17GB |
1from transformers import GPT2Tokenizer, GPT2LMHeadModel
2
3def generate_response(input_text):
4
5 inputs = tokenizer(input_text, return_tensors="pt")
6 output_sequences = model.generate(
7 input_ids=inputs['input_ids'],
8 attention_mask=inputs['attention_mask'],
9 max_length=100, # Adjusted max_length
10 temperature=0.3,
11 top_k=40,
12 top_p=0.85,
13 num_return_sequences=1,
14 no_repeat_ngram_size=2,
15 pad_token_id=tokenizer.eos_token_id,
16 eos_token_id=tokenizer.eos_token_id,
17 early_stopping=True,
18 do_sample=True,
19 use_cache=True,
20 )
21
22 full_generated_text = tokenizer.decode(output_sequences[0], skip_special_tokens=True)
23
24 bot_response_start = full_generated_text.find('[Bot]') + len('[Bot]')
25 bot_response = full_generated_text[bot_response_start:]
26 return bot_response
27
28
29model_name = 'KhantKyaw/Chat_GPT-2'
30tokenizer = GPT2Tokenizer.from_pretrained(model_name)
31model = GPT2LMHeadModel.from_pretrained(model_name)
32response = generate_response(user_input)
33print("Chatbot:", response)
34