スポンサーリンク
以前にgoogle colab上でdiffusersを動かして、画像生成を行いました
technoxs-stacker.hatenablog.com
diffusersモジュールを使う際、seed値を指定しないとランダムに画像が生成されます
この状態だと、毎回異なる画像が生成されてしまいます
同じ画像が生成できるようにしておくと、プロンプトやパラメータ調整しやすいです
今回はseed値を外部から指定して、同じ画像を繰り返し生成してみます
目次
スポンサーリンク
この記事でわかること
diffusersでシード値を固定して画像生成する方法
1.実行環境
Google Colab
Diffusers
transformers==4.26.0
model : chilloutmix_NiPrunedFp32Fix
lora : koreanDollLikeness_v15
lora:Nakano nino in The Quintessential Quintuplets | Realistic LORA
2.画像生成コード
model_name = "nuigurumi/basil_mix" save_name = 'basil_mix.png' save_dir = path/to/save/directory os.makedirs(save_dir, exist_ok=True) save_path = os.path.join(save_dir, save_name) seed = 3000000 device = "cpu" pipe = StableDiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.float32) pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config) pipe = pipe.to("cuda") if pipe.safety_checker is not None: pipe.safety_checker = lambda images, **kwargs: (images, False) positive="masterpiece, ultra high res, 1 girl, (photo realistic:1.2),small_breasts,cleavage,smile,detailed beautiful skin, face focus, (black eyes), cute, young, looking at viewers, (scoop neck tee:2.0), ,collarbone, ((black hair)), windy, soft light, hand between legs, from front" negative="painting,sketches,(worst quality:2),(low quality:2),(normal quality:2),lowers,normal quality,((monochrome)),((grayscale)),skin spots,acnes,skin blemishes,age spot,(outdoor:1.6),nsfw,ugly face,fat,(missing fingers),extra fingers,extra arms,extra legs,open chest,chest pad" generator = torch.Generator(device).manual_seed(seed) image = pipe(positive, negative_prompt=negative, generator=generator).images[0] image.save(save_path)
3.生成結果
positive
masterpiece ultra_highres 1girl (photo realistic:2) (detailed beautiful skin face focus) (extremely detailed eyes and face) (extremely detailed CG) cute (full_body_shot) looking_at_viewers (white_shirt) green_skirt pleated_skirt long_hair (butterfly hair ornament) (dark_pink_hair:2)
negative
sketches, (worst quality:2), (low quality:2), (normal quality:2), lowres, normal quality, ((monochrome)), ((grayscale)), skin spots, acnes, skin blemishes, bad anatomy,(long hair:1.4),DeepNegative,(fat:1.2),facing away, looking away,tilted head, lowres,bad anatomy,bad hands, text, error, missing fingers,extra digit, fewer digits, cropped, worstquality, low quality, normal quality,jpegartifacts,signature, watermark, username,blurry,bad feet,cropped,poorly drawn hands,poorly drawn face,mutation,deformed,worst quality,low quality,normal quality,jpeg artifacts,signature,watermark,extra fingers,fewer digits,extra limbs,extra arms,extra legs,malformed limbs,fused fingers,too many fingers,long neck,cross-eyed,mutated hands,polar lowres,bad body,bad proportions,gross proportions,text,error,missing fingers,missing arms,missing legs,extra digit, extra arms, extra leg, extra foot
seed = 6763306306178
seed = 6763306306178
スポンサーリンク
deeplearning関連記事
technoxs-stacker.hatenablog.com technoxs-stacker.hatenablog.com technoxs-stacker.hatenablog.com
参考
murasan-net.com huggingface.co github.com yuuyuublog.org
diffusers関連記事
別記事でコスプレ画像を作成してます~
modelやLoRA、プロンプトの参考になればうれしいです(^^)/
technoxs-stacker.hatenablog.com
technoxs-stacker.hatenablog.com
technoxs-stacker.hatenablog.com