Update README.md
Browse files
README.md
CHANGED
|
@@ -49,9 +49,9 @@ huggingface-cli download --resume-download UnicomAI/UniT2IXL
|
|
| 49 |
|
| 50 |
3. 推理`demo.py`:
|
| 51 |
```bash
|
| 52 |
-
from
|
| 53 |
-
|
| 54 |
-
|
| 55 |
prompt = "在绿色的森林中,隐藏着一座白色的哥特式教堂,教堂的尖塔直指蓝色的天空,教堂周围是五彩斑斓的野花和浅黄色的草坪。"
|
| 56 |
-
image =
|
| 57 |
```
|
|
|
|
| 49 |
|
| 50 |
3. 推理`demo.py`:
|
| 51 |
```bash
|
| 52 |
+
from pipeline_unit2ixl import UniT2IXLPipeline
|
| 53 |
+
pipeline = UniT2IXLPipeline.from_pretrained("UnicomAI/UniT2IXL")
|
| 54 |
+
pipeline = pipeline.to("cuda:0")
|
| 55 |
prompt = "在绿色的森林中,隐藏着一座白色的哥特式教堂,教堂的尖塔直指蓝色的天空,教堂周围是五彩斑斓的野花和浅黄色的草坪。"
|
| 56 |
+
image = pipeline(prompt=prompt,guidance_scale=7.5,target_size=(1024,1024)).images[0]
|
| 57 |
```
|