darkps commited on
Commit
1aaed88
·
verified ·
1 Parent(s): a360add

Upload notebook.ipynb

Browse files
Files changed (1) hide show
  1. notebook.ipynb +19 -19
notebook.ipynb CHANGED
@@ -19,21 +19,21 @@
19
  "from llama_cpp import Llama\n",
20
  "import os\n",
21
  "\n",
22
- "REPO_ID = \"darkps/darkit-v2.5\"\n",
23
- "api = HfApi()\n",
24
  "\n",
25
- "files = api.list_repo_files(REPO_ID)\n",
26
- "gguf_files = [f for f in files if f.endswith(\".gguf\")]\n",
27
  "\n",
28
- "for i, f in enumerate(gguf_files):\n",
29
- " print(f\"[{i}] {f}\")\n",
30
  "\n",
31
- "choice = int(input(\"Select model number: \"))\n",
32
- "filename = gguf_files[choice]\n",
33
  "\n",
34
- "llm = Llama.from_pretrained(\n",
35
- " repo_id=REPO_ID,\n",
36
- " filename=filename,\n",
37
  " n_ctx=2048,\n",
38
  " n_batch=128,\n",
39
  " n_ubatch=128,\n",
@@ -51,12 +51,12 @@
51
  "metadata": {},
52
  "outputs": [],
53
  "source": [
54
- "llm.set_cache(None)\n",
55
  "\n",
56
- "PROMPT = \"Hi how are you?\"\n",
57
  "\n",
58
- "stream = llm(\n",
59
- " f\"<|im_start|>user\\n{PROMPT}<|im_end|>\\n<|im_start|>assistant\\n\",\n",
60
  " max_tokens=128,\n",
61
  " temperature=0.7,\n",
62
  " top_p=0.8,\n",
@@ -71,11 +71,11 @@
71
  " echo=False\n",
72
  ")\n",
73
  "\n",
74
- "for chunk in stream:\n",
75
- " text = chunk[\"choices\"][0][\"text\"]\n",
76
  "\n",
77
- " if text:\n",
78
- " print(text, end=\"\", flush=True)\n",
79
  "\n",
80
  "print()\n"
81
  ]
 
19
  "from llama_cpp import Llama\n",
20
  "import os\n",
21
  "\n",
22
+ "ИДЕНТИФИКАТОР_РЕПО = \"darkps/darkit-v2.5\"\n",
23
+ "АПИ = HfApi()\n",
24
  "\n",
25
+ "ФАЙЛЫ = АПИ.list_repo_files(ИДЕНТИФИКАТОР_РЕПО)\n",
26
+ "ФАЙЛЫ_GGUF = [файл for файл in ФАЙЛЫ if файл.endswith(\".gguf\")]\n",
27
  "\n",
28
+ "for индекс, файл in enumerate(ФАЙЛЫ_GGUF):\n",
29
+ " print(f\"[{индекс}] {файл}\")\n",
30
  "\n",
31
+ "ВЫБОР = int(input(\"Выберите номер модели: \"))\n",
32
+ "ИМЯ_ФАЙЛА = ФАЙЛЫ_GGUF[ВЫБОР]\n",
33
  "\n",
34
+ "МОДЕЛЬ = Llama.from_pretrained(\n",
35
+ " repo_id=ИДЕНТИФИКАТОР_РЕПО,\n",
36
+ " filename=ИМЯ_ФАЙЛА,\n",
37
  " n_ctx=2048,\n",
38
  " n_batch=128,\n",
39
  " n_ubatch=128,\n",
 
51
  "metadata": {},
52
  "outputs": [],
53
  "source": [
54
+ "МОДЕЛЬ.set_cache(None)\n",
55
  "\n",
56
+ "ЗАПРОС = \"Привет, как у тебя дела?\"\n",
57
  "\n",
58
+ "ПОТОК = МОДЕЛЬ(\n",
59
+ " f\"<|im_start|>user\\n{ЗАПРОС}<|im_end|>\\n<|im_start|>assistant\\n\",\n",
60
  " max_tokens=128,\n",
61
  " temperature=0.7,\n",
62
  " top_p=0.8,\n",
 
71
  " echo=False\n",
72
  ")\n",
73
  "\n",
74
+ "for ЧАСТЬ in ПОТОК:\n",
75
+ " ТЕКСТ = ЧАСТЬ[\"choices\"][0][\"text\"]\n",
76
  "\n",
77
+ " if ТЕКСТ:\n",
78
+ " print(ТЕКСТ, end=\"\", flush=True)\n",
79
  "\n",
80
  "print()\n"
81
  ]