import argparse
import torch
from diffusers import DiffusionPipeline
import os
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='arg parser')
parser.add_argument('-i', '--input', required=True, type=argparse.FileType('r'))
args = parser.parse_args()
queries = args.input.readlines()
model = DiffusionPipeline.from_pretrained(
"stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True
)
model.to("cuda")
if not os.path.exists('images'):
os.makedirs('images')
i = 0
for prompt in queries:
print(f'готовимся генерировать картинку')
image = model(prompt=prompt).images[0]
image.save(f"images/{i}-base.png")
i += 1
Или можно поставить Автоматик A1111
Извиняюсь, а чем это лучше варианта использовать готовый automatic1111?(ну и вообще других фронтендов)
Там просто куча всяких полезных плюшек вместе с удобный гуи, поддержка лор, удобно сейвить все, батчи, инпейнты
так и тут можно) у тебя есть код - делай что хочешь, какие угодно модели, какие угодно пред\пост обработки
Но в Автоматике есть api..
ну да, но тут статья скорее про то как это все сделать самому из кода
там выше в комментах подали прикольную идею про расписывание как к этому подключить всякие лоры и инпейнты - возможно я ее даже напишу
Я в ахуе. Нейродебилы про автоматик для васянов не знают?
так там дисклеймер что статья не для васянов)