ahmedfaiyaz
/

Mukh-Oboyob

MukhOboyobPipeline

stable-diffusion

Model card Files Files and versions Community

Ahmed Faiyaz commited on Jan 12, 2024

Commit

0b3569f

•

1 Parent(s): 63e99ed

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ inference:
 from diffusers import DiffusionPipeline
 from transformers import AutoTokenizer,AutoModel
 from diffusers.models import AutoencoderKL
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-ema")
 pipeline = DiffusionPipeline.from_pretrained(

 from diffusers import DiffusionPipeline
 from transformers import AutoTokenizer,AutoModel
 from diffusers.models import AutoencoderKL
+device="cpu"
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-ema")
 pipeline = DiffusionPipeline.from_pretrained(