Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,110 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
language: en
|
3 |
+
license: apache-2.0
|
4 |
+
tags:
|
5 |
+
- art
|
6 |
+
- image-generation
|
7 |
+
- text-to-image
|
8 |
+
- diffusion
|
9 |
+
- high-quality
|
10 |
+
- AI
|
11 |
+
pipeline_tag: text-to-image
|
12 |
+
thumbnail: https://example.com/thumbnail.png
|
13 |
+
---
|
14 |
+
|
15 |
+
# ImageGroundV0.1
|
16 |
+
|
17 |
+
**Model Card for `enhanceaiteam/ImageGroundV0.1`**
|
18 |
+
|
19 |
+
`ImageGroundV0.1` is a powerful text-to-image generation model developed by EnhanceAI. It leverages advanced diffusion techniques to produce high-quality, detailed images based on textual descriptions. This model is designed to assist artists, designers, and creative professionals in generating visually compelling and accurate images.
|
20 |
+
|
21 |
+
## Model Details
|
22 |
+
|
23 |
+
- **Version**: 0.1
|
24 |
+
- **Base Model**: EnhanceAI's proprietary image generation architecture
|
25 |
+
- **Training Data**: The model has been trained on a diverse dataset of high-resolution, copyright-free images, ensuring a broad range of visual styles and subjects.
|
26 |
+
- **Pipeline**: Stable Diffusion XL Pipeline
|
27 |
+
|
28 |
+
## Features
|
29 |
+
|
30 |
+
- **High-Resolution Output**: Generates images at 1024x1024 resolution.
|
31 |
+
- **Enhanced Prompt Understanding**: Improved comprehension of complex and detailed text prompts.
|
32 |
+
- **Versatile Style Generation**: Capable of producing images in various artistic and realistic styles.
|
33 |
+
- **High-Quality Images**: Delivers visually stunning and detailed images.
|
34 |
+
|
35 |
+
## Usage
|
36 |
+
|
37 |
+
To use `ImageGroundV0.1`, you can integrate it with the `diffusers` library. Below is an example of how to generate images:
|
38 |
+
|
39 |
+
```python
|
40 |
+
import torch
|
41 |
+
from diffusers import (
|
42 |
+
StableDiffusionXLPipeline,
|
43 |
+
KDPM2AncestralDiscreteScheduler,
|
44 |
+
AutoencoderKL
|
45 |
+
)
|
46 |
+
|
47 |
+
# Load VAE component
|
48 |
+
vae = AutoencoderKL.from_pretrained(
|
49 |
+
"madebyollin/sdxl-vae-fp16-fix",
|
50 |
+
torch_dtype=torch.float16
|
51 |
+
)
|
52 |
+
|
53 |
+
# Configure the pipeline
|
54 |
+
pipe = StableDiffusionXLPipeline.from_pretrained(
|
55 |
+
"enhanceaiteam/ImageGroundV0.1",
|
56 |
+
vae=vae,
|
57 |
+
torch_dtype=torch.float16
|
58 |
+
)
|
59 |
+
pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
60 |
+
pipe.to('cuda')
|
61 |
+
|
62 |
+
# Define prompts and generate image
|
63 |
+
prompt = "a serene landscape with mountains and a clear sky"
|
64 |
+
negative_prompt = ""
|
65 |
+
|
66 |
+
image = pipe(
|
67 |
+
prompt,
|
68 |
+
negative_prompt=negative_prompt,
|
69 |
+
width=1024,
|
70 |
+
height=1024,
|
71 |
+
guidance_scale=7,
|
72 |
+
num_inference_steps=50,
|
73 |
+
clip_skip=2
|
74 |
+
).images[0]
|
75 |
+
|
76 |
+
image.save("generated_image.png")
|
77 |
+
```
|
78 |
+
|
79 |
+
## Prompts
|
80 |
+
|
81 |
+
When creating prompts, be as descriptive as possible to achieve the desired results. Detailed descriptions help the model generate more accurate and relevant images.
|
82 |
+
|
83 |
+
## Image Samples
|
84 |
+
|
85 |
+
Here are some example images generated by `ImageGroundV0.1`:
|
86 |
+
|
87 |
+
- 
|
88 |
+
- 
|
89 |
+
- 
|
90 |
+
- 
|
91 |
+
- 
|
92 |
+
- 
|
93 |
+
|
94 |
+
## Explore
|
95 |
+
|
96 |
+
Explore and try out `ImageGroundV0.1` on the [EnhanceAI Playground](https://enhanceai.art/playground). Experiment with different prompts and see the incredible results you can achieve.
|
97 |
+
|
98 |
+
## License
|
99 |
+
|
100 |
+
`ImageGroundV0.1` is licensed under the [Apache 2.0 License](https://opensource.org/licenses/Apache-2.0).
|
101 |
+
|
102 |
+
## Contact
|
103 |
+
|
104 |
+
For support and inquiries, please reach out to [[email protected]](mailto:[email protected]).
|
105 |
+
|
106 |
+
---
|
107 |
+
|
108 |
+
Happy creating!
|
109 |
+
|
110 |
+
- The EnhanceAI Team
|