用C语言写了个Lora对话集生成器

苍生墨客

已于 2023-10-23 16:49:34 修改

阅读量110

点赞数 1

文章标签： c语言开发语言

于 2023-10-23 16:42:13 首次发布

本文链接：https://blog.csdn.net/2301_79077750/article/details/133993495

版权

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<stdlib.h>
int main()
{
	int n = 1000,choose=1,choose1=0,a=1;
	char *io = (char*)malloc(sizeof(char)*n);
	FILE* pr= fopen("2.json", "a");
	if (pr == NULL)
	{
		printf("失败");
	}
	else
	{
		while (1)
		{
			printf("问：");
			scanf("%s", io);
			if (a == 1)
			{
				fprintf(pr, "[");
			}
			fprintf(pr, "\n  {\n    \"instruction\":\"%s\",\n    \"input\": \"\",\n", io);
			printf("答：");
			scanf("%s", io);
			printf("继续按1，否则按0：");
			scanf("%d", &choose);
			if (choose == 1)
			{
				printf("是否引入历史对话：是则输入1，否则输入其它");
				scanf("%d", &choose1);
				if (choose1 == 1)
				{
					fprintf(pr, "    \"output\":\"%s\",\n    \"history\": [\n", io);
					printf("历史问：");
					scanf("%s", io);
					fprintf(pr, "      [\n        \"%s\",\n", io);
					printf("历史答：");
					scanf("%s", io);
					fprintf(pr, "        \"%s\"\n      ]\n    ]\n  },", io);
					printf("新一轮对话：\n");
				}
				else
				{
					fprintf(pr, "    \"output\":\"%s\",\n    \"history\": []\n  },", io);
				}
			}
			else
			{
				fprintf(pr, "    \"output\":\"%s\",\n    \"history\": []\n  }\n]", io);
				break;
			}
			a++;
		}
		fclose(pr);
	}
    free(io);
	return 0;
}

白嫖了阿里云三个月的v100，就去学了在dsw里部署glm2-6b，之后又学了llma微调，觉得复制粘贴太麻烦干脆自己写个lora对话集生成器；

有个问题，生成的json文件直接替换llma提供的样例JSON文件就无法识别，但用生成的JSON文件里的内容替换样例JSON文件里的内容就可以识别了，不清楚什么情况。

刚入大学的萌新求大佬解惑😭；