modelscope usage
pip install invisible_watermark transformers safetensors
pip install diffusers==0.18.0
from modelscope.utils.constant import Tasks
from modelscope.pipelines import pipeline
import cv2
pipe = pipeline(task=Tasks.text_to_image_synthesis,
model='NickYuan/XLMoyouArtificial_v01',
use_safetensors=True,
model_revision='v1.0.0')
prompt = 'a dog'
output = pipe({'text': prompt})
cv2.imwrite('result.png', output['output_imgs'][0])
Clone with HTTP
git clone https://www.modelscope.cn/NickYuan/XLMoyouArtificial_v01.git
评论