Update README.md

README.md

@@ -14,19 +14,19 @@ instance_prompt: null
 
 ---
 # Mukh-Oboyob
+## Usage
 ```py
-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(
-      "CompVis/stable-diffusion-v1-4",
-      tokenizer=AutoTokenizer.from_pretrained('csebuetnlp/banglabert'),
-      text_encoder = AutoModel.from_pretrained("csebuetnlp/banglabert"),
-      vae=vae
-)
+import torch
+from diffusers import StableDiffusionPipeline
+
+model_id = "CompVis/stable-diffusion-v1-4"
+device = "cuda"
+
+
+pipe = StableDiffusionPipeline.from_pretrained(model_id)
+pipe.unet.load_attn_procs("gr33nr1ng3r/Mukh-Oboyob")
+pipe = pipe.to(device)
+
 ```
 <Gallery />