We previously ran diffusers on google colab to generate images.
technoxs-stacker.hatenablog.com
This article is a memo to use the diffusers module with google colab.
contents
スポンサーリンク
abstract
How to build a diffusers environment on google colab
1.requirement
Google Colab
Diffusers
transformers==4.26.0
model : basil_mix
2.install
!pip install diffusers transformers==4.26.0 ftfy accelerate !pip install safetensors
3.code
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)
4.result
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
seed = 60000
seed = 90000
seed = 9000000
スポンサーリンク
deeplearning related articles
technoxs-stacker.hatenablog.com technoxs-stacker.hatenablog.com technoxs-stacker.hatenablog.com