commit 2a3d9d851d53ac5702d305db01dbc5c99d1fd8a2 Author: ModelHub XC Date: Sat Apr 25 22:58:55 2026 +0800 初始化项目，由ModelHub XC社区提供模型 Model: pai/pai-bloom-1b1-text2prompt-sd Source: Original Platform diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..21b3632 --- /dev/null +++ b/.gitattributes @@ -0,0 +1,49 @@ +*.7z filter=lfs diff=lfs merge=lfs -text +*.arrow filter=lfs diff=lfs merge=lfs -text +*.bin filter=lfs diff=lfs merge=lfs -text +*.bin.* filter=lfs diff=lfs merge=lfs -text +*.bz2 filter=lfs diff=lfs merge=lfs -text +*.ftz filter=lfs diff=lfs merge=lfs -text +*.gz filter=lfs diff=lfs merge=lfs -text +*.h5 filter=lfs diff=lfs merge=lfs -text +*.joblib filter=lfs diff=lfs merge=lfs -text +*.lfs.* filter=lfs diff=lfs merge=lfs -text +*.model filter=lfs diff=lfs merge=lfs -text +*.msgpack filter=lfs diff=lfs merge=lfs -text +*.onnx filter=lfs diff=lfs merge=lfs -text +*.ot filter=lfs diff=lfs merge=lfs -text +*.parquet filter=lfs diff=lfs merge=lfs -text +*.pb filter=lfs diff=lfs merge=lfs -text +*.pt filter=lfs diff=lfs merge=lfs -text +*.pth filter=lfs diff=lfs merge=lfs -text +*.rar filter=lfs diff=lfs merge=lfs -text +saved_model/**/* filter=lfs diff=lfs merge=lfs -text +*.tar.* filter=lfs diff=lfs merge=lfs -text +*.tflite filter=lfs diff=lfs merge=lfs -text +*.tgz filter=lfs diff=lfs merge=lfs -text +*.xz filter=lfs diff=lfs merge=lfs -text +*.zip filter=lfs diff=lfs merge=lfs -text +*.zstandard filter=lfs diff=lfs merge=lfs -text +*.tfevents* filter=lfs diff=lfs merge=lfs -text +*.db* filter=lfs diff=lfs merge=lfs -text +*.ark* filter=lfs diff=lfs merge=lfs -text +**/*ckpt*data* filter=lfs diff=lfs merge=lfs -text +**/*ckpt*.meta filter=lfs diff=lfs merge=lfs -text +**/*ckpt*.index filter=lfs diff=lfs merge=lfs -text +*.safetensors filter=lfs diff=lfs merge=lfs -text +*.ckpt filter=lfs diff=lfs merge=lfs -text +*.gguf* filter=lfs diff=lfs merge=lfs -text +*.ggml filter=lfs diff=lfs merge=lfs -text +*.llamafile* filter=lfs diff=lfs merge=lfs -text +*.pt2 filter=lfs diff=lfs merge=lfs -text +*.mlmodel filter=lfs diff=lfs merge=lfs -text +*.npy filter=lfs diff=lfs merge=lfs -text +*.npz filter=lfs diff=lfs merge=lfs -text +*.pickle filter=lfs diff=lfs merge=lfs -text +*.pkl filter=lfs diff=lfs merge=lfs -text +*.tar filter=lfs diff=lfs merge=lfs -text +*.wasm filter=lfs diff=lfs merge=lfs -text +*.zst filter=lfs diff=lfs merge=lfs -text +*tfevents* filter=lfs diff=lfs merge=lfs -text + +tokenizer.json filter=lfs diff=lfs merge=lfs -text \ No newline at end of file diff --git a/README.md b/README.md new file mode 100644 index 0000000..1cc36fd --- /dev/null +++ b/README.md @@ -0,0 +1,98 @@ +--- +license: apache-2.0 +widget: + - text: "Instruction: Give a simple description of the image to generate a drawing prompt.\nInput: 1 girl\nOutput:" +tags: +- pytorch +- transformers +- text-generation +--- + +# BeautifulPrompt + +## 简介 Brief Introduction + +我们开源了一个自动Prompt生成模型，您可以直接输入一个极其简单的Prompt，就可以得到经过语言模型优化过的Prompt，帮助您更简单地生成高颜值图像。 + +We release an automatic Prompt generation model, you can directly enter an extremely simple Prompt and get a Prompt optimized by the language model to help you generate more beautiful images simply. + +* Github: [EasyNLP](https://github.com/alibaba/EasyNLP) + +## 使用 Usage + +```python +from transformers import AutoTokenizer, AutoModelForCausalLM + +tokenizer = AutoTokenizer.from_pretrained('alibaba-pai/pai-bloom-1b1-text2prompt-sd') +model = AutoModelForCausalLM.from_pretrained('alibaba-pai/pai-bloom-1b1-text2prompt-sd').eval().cuda() + +raw_prompt = '1 girl' +input = f'Instruction: Give a simple description of the image to generate a drawing prompt.\nInput: {raw_prompt}\nOutput:' +input_ids = tokenizer.encode(input, return_tensors='pt').cuda() + +outputs = model.generate( + input_ids, + max_length=384, + do_sample=True, + temperature=1.0, + top_k=50, + top_p=0.95, + repetition_penalty=1.2, + num_return_sequences=5) + +prompts = tokenizer.batch_decode(outputs[:, input_ids.size(1):], skip_special_tokens=True) +prompts = [p.strip() for p in prompts] +print(prompts) +``` + +## 作品展示 Gallery + + + +| Original | BeautifulPrompt | +| ---------------------------------------- | ---------------------------------- | +| prompt: taylor swift, country, golden, fearless,wavehair | prompt: portrait of taylor swift as a beautiful woman, long hair, country, golden ratio, intricate, symmetrical, cinematic lighting, highly detailed, digital painting, artstation, concept art, smooth, sharp focus, illustration | +| ![](example1.png) | ![](example2.png) | + + +| Original | BeautifulPrompt | +| ---------------------------------------- | ---------------------------------- | +| prompt: A majestic sailing ship | prompt: a massive sailing ship, epic, cinematic, artstation, greg rutkowski, james gurney, sparth | +| ![](example3.png) | ![](example4.png) | + + + +## 使用须知 Notice for Use + +使用上述模型需遵守[AIGC模型开源特别条款](https://terms.alicdn.com/legal-agreement/terms/common_platform_service/20230505180457947/20230505180457947.html)。 + +If you want to use this model, please read this [document](https://terms.alicdn.com/legal-agreement/terms/common_platform_service/20230505180457947/20230505180457947.html) carefully and abide by the terms. + +## Paper Citation + +If you find the model useful, please consider cite the paper: + +``` +@inproceedings{emnlp2023a, + author = {Tingfeng Cao and + Chengyu Wang and + Bingyan Liu and + Ziheng Wu and + Jinhui Zhu and + Jun Huang}, + title = {BeautifulPrompt: Towards Automatic Prompt Engineering for Text-to-Image Synthesis}, + booktitle = {Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing: Industry Track}, + pages = {1--11}, + year = {2023} +} +``` + + + diff --git a/config.json b/config.json new file mode 100644 index 0000000..d8c2327 --- /dev/null +++ b/config.json @@ -0,0 +1,32 @@ +{ + "_name_or_path": "alibaba-pai/pai-bloom-1b1-text2prompt-sd", + "apply_residual_connection_post_layernorm": false, + "architectures": [ + "BloomForCausalLM" + ], + "attention_dropout": 0.0, + "attention_softmax_in_fp32": true, + "bias_dropout_fusion": true, + "bos_token_id": 1, + "eos_token_id": 2, + "hidden_dropout": 0.0, + "hidden_size": 1536, + "initializer_range": 0.02, + "layer_norm_epsilon": 1e-05, + "masked_softmax_fusion": true, + "model_type": "bloom", + "n_head": 16, + "n_inner": null, + "n_layer": 24, + "offset_alibi": 100, + "pad_token_id": 3, + "pretraining_tp": 1, + "skip_bias_add": true, + "skip_bias_add_qkv": false, + "slow_but_exact": false, + "torch_dtype": "float16", + "transformers_version": "4.27.4", + "unk_token_id": 0, + "use_cache": true, + "vocab_size": 250880 +} diff --git a/configuration.json b/configuration.json new file mode 100644 index 0000000..bbeeda1 --- /dev/null +++ b/configuration.json @@ -0,0 +1 @@ +{"framework": "pytorch", "task": "text-generation", "allow_remote": true} \ No newline at end of file diff --git a/example1.png b/example1.png new file mode 100644 index 0000000..5573825 Binary files /dev/null and b/example1.png differ diff --git a/example2.png b/example2.png new file mode 100644 index 0000000..2f9d47c Binary files /dev/null and b/example2.png differ diff --git a/example3.png b/example3.png new file mode 100644 index 0000000..9928d46 Binary files /dev/null and b/example3.png differ diff --git a/example4.png b/example4.png new file mode 100644 index 0000000..d80608f Binary files /dev/null and b/example4.png differ diff --git a/generation_config.json b/generation_config.json new file mode 100644 index 0000000..c29e337 --- /dev/null +++ b/generation_config.json @@ -0,0 +1,7 @@ +{ + "_from_model_config": true, + "bos_token_id": 1, + "eos_token_id": 2, + "pad_token_id": 3, + "transformers_version": "4.29.2" +} diff --git a/model.safetensors b/model.safetensors new file mode 100644 index 0000000..71a54f8 --- /dev/null +++ b/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0d19a7a6ac4065977abdd26715cead6b2927853ac1f1f788c824d0e88b51d1e1 +size 2130662600 diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..fe35a54 --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:f8da7879590b07ed703a20a150bb3dd20e6df6d132596827c2b7b5c1570a32cd +size 2130723617 diff --git a/special_tokens_map.json b/special_tokens_map.json new file mode 100644 index 0000000..09433d5 --- /dev/null +++ b/special_tokens_map.json @@ -0,0 +1,7 @@ +{ + "bos_token": "", + "eos_token": "", + "pad_token": "", + "sep_token": "", + "unk_token": "" +} diff --git a/tokenizer.json b/tokenizer.json new file mode 100644 index 0000000..dbf002c --- /dev/null +++ b/tokenizer.json @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:17a208233d2ee8d8c83b23bc214df737c44806a1919f444e89b31e586cd956ba +size 14500471 diff --git a/tokenizer_config.json b/tokenizer_config.json new file mode 100644 index 0000000..612d97d --- /dev/null +++ b/tokenizer_config.json @@ -0,0 +1,11 @@ +{ + "add_prefix_space": false, + "bos_token": "", + "clean_up_tokenization_spaces": false, + "eos_token": "", + "model_max_length": 1000000000000000019884624838656, + "pad_token": "", + "padding_side": "left", + "tokenizer_class": "BloomTokenizer", + "unk_token": "" +}