Safetensors
Aiwensile2 commited on
Commit
126a11c
·
verified ·
1 Parent(s): face319

Delete Usage guidelines available for reference.ipynb

Browse files
Usage guidelines available for reference.ipynb DELETED
@@ -1,136 +0,0 @@
1
- {
2
- "cells": [
3
- {
4
- "cell_type": "code",
5
- "execution_count": null,
6
- "id": "a3a51bb2-faa9-47ef-8cc2-c33761bc16b6",
7
- "metadata": {},
8
- "outputs": [],
9
- "source": [
10
- "# Complete model usage guidelines available for reference, applicable to the MineMA-8B series models\n",
11
- "from transformers import AutoTokenizer, AutoModelForCausalLM\n",
12
- "import torch\n",
13
- "\n",
14
- "model_id = \"\" # model path\n",
15
- "\n",
16
- "tokenizer = AutoTokenizer.from_pretrained(model_id)\n",
17
- "model = AutoModelForCausalLM.from_pretrained(\n",
18
- " model_id,\n",
19
- " torch_dtype=torch.bfloat16,\n",
20
- " device_map = {\"\": 0},\n",
21
- ")\n",
22
- "\n",
23
- "sys_mes = \"\"\n",
24
- "user_mes = \"\"\"\n",
25
- "\n",
26
- "\"\"\"\n",
27
- "\n",
28
- "messages = [\n",
29
- " {\"role\": \"system\", \"content\": sys_mes},\n",
30
- " {\"role\": \"user\", \"content\": user_mes},\n",
31
- "]\n",
32
- "\n",
33
- "input_ids = tokenizer.apply_chat_template(\n",
34
- " messages,\n",
35
- " add_generation_prompt=True,\n",
36
- " return_tensors=\"pt\"\n",
37
- ").to(model.device)\n",
38
- "\n",
39
- "terminators = [\n",
40
- " tokenizer.eos_token_id,\n",
41
- " tokenizer.convert_tokens_to_ids(\"<|eot_id|>\")\n",
42
- "]\n",
43
- "\n",
44
- "outputs = model.generate(\n",
45
- " input_ids,\n",
46
- " max_new_tokens=256,\n",
47
- " eos_token_id=terminators,\n",
48
- " do_sample=True,\n",
49
- " temperature=0.6,\n",
50
- " top_p=0.9,\n",
51
- " #repetition_penalty=1.3, \n",
52
- ")\n",
53
- "response = outputs[0][input_ids.shape[-1]:]\n",
54
- "print(tokenizer.decode(response, skip_special_tokens=True))"
55
- ]
56
- },
57
- {
58
- "cell_type": "code",
59
- "execution_count": null,
60
- "id": "634637b4-e04f-4d3a-a77e-712e53617249",
61
- "metadata": {},
62
- "outputs": [],
63
- "source": [
64
- "# LoRA model usage guidelines available for reference, applicable to the MineMA-70B series models\n",
65
- "from transformers import AutoModelForCausalLM, AutoTokenizer\n",
66
- "import torch\n",
67
- "import json\n",
68
- "from peft import PeftModel, LoraConfig, TaskType\n",
69
- "\n",
70
- "mode_path = '' # base model path\n",
71
- "lora_path = '' # lora model path\n",
72
- "\n",
73
- "tokenizer = AutoTokenizer.from_pretrained(mode_path)\n",
74
- "\n",
75
- "model = AutoModelForCausalLM.from_pretrained(mode_path, device_map=\"auto\",torch_dtype=torch.bfloat16)\n",
76
- "\n",
77
- "with open('config.json', 'r') as f:\n",
78
- " lora_config_data = json.load(f)\n",
79
- "\n",
80
- "config = LoraConfig(**lora_config_data)\n",
81
- "\n",
82
- "model = PeftModel.from_pretrained(model, model_id=lora_path, config=config)\n",
83
- "\n",
84
- "prompt = \"How to get diamond in Minecraft?\"\n",
85
- "messages = [\n",
86
- " {\"role\": \"system\", \"content\": \"You are a Large Language Model, and your task is to answer questions posed by users about Minecraft. Utilize your knowledge and understanding of the game to provide detailed, accurate, and helpful responses. Use your capabilities to assist users in solving problems, understanding game mechanics, and enhancing their Minecraft experience.\"},\n",
87
- " {\"role\": \"user\", \"content\":prompt}\n",
88
- "]\n",
89
- "\n",
90
- "input_ids = tokenizer.apply_chat_template(\n",
91
- " messages,\n",
92
- " add_generation_prompt=True,\n",
93
- " return_tensors=\"pt\"\n",
94
- ").to(model.device)\n",
95
- "\n",
96
- "terminators = [\n",
97
- " tokenizer.eos_token_id,\n",
98
- " tokenizer.convert_tokens_to_ids(\"<|eot_id|>\")\n",
99
- "]\n",
100
- "\n",
101
- "outputs = model.generate(\n",
102
- " input_ids,\n",
103
- " max_new_tokens=256,\n",
104
- " eos_token_id=terminators,\n",
105
- " do_sample=True,\n",
106
- " temperature=0.6,\n",
107
- " top_p=0.9,\n",
108
- " #repetition_penalty=1.3, \n",
109
- ")\n",
110
- "response = outputs[0][input_ids.shape[-1]:]\n",
111
- "print(tokenizer.decode(response, skip_special_tokens=True))"
112
- ]
113
- }
114
- ],
115
- "metadata": {
116
- "kernelspec": {
117
- "display_name": "Python 3 (ipykernel)",
118
- "language": "python",
119
- "name": "python3"
120
- },
121
- "language_info": {
122
- "codemirror_mode": {
123
- "name": "ipython",
124
- "version": 3
125
- },
126
- "file_extension": ".py",
127
- "mimetype": "text/x-python",
128
- "name": "python",
129
- "nbconvert_exporter": "python",
130
- "pygments_lexer": "ipython3",
131
- "version": "3.9.7"
132
- }
133
- },
134
- "nbformat": 4,
135
- "nbformat_minor": 5
136
- }