Ahmed Faiyaz commited on
Commit
238a946
·
verified ·
1 Parent(s): f6d06b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -3,14 +3,13 @@ import torch
3
  from diffusers import DiffusionPipeline
4
  from transformers import AutoTokenizer,AutoModel
5
  from diffusers.models import AutoencoderKL
6
- device="cuda"
7
  pipeline = DiffusionPipeline.from_pretrained(
8
  "CompVis/stable-diffusion-v1-4",
9
  text_encoder = AutoModel.from_pretrained("csebuetnlp/banglabert"),
10
  custom_pipeline="gr33nr1ng3r/Mukh-Oboyob"
11
  )
12
  pipeline.unet.load_attn_procs("gr33nr1ng3r/Mukh-Oboyob")
13
- pipeline.to(device)
14
 
15
  def diffusion(text,num_inference_steps,guidance_scale):
16
  prompt = "মেয়েটির কালো চুল ছিল। মেয়েটির মুখে ভারী মেকাপ ছিল। মেয়েটির উঁচু গালের হাড় ছিল। মেয়েটির মুখ কিছুটা খোলা ছিল। মেয়েটির চেহারা ডিম্বাকৃতির। মেয়েটির চোখা নাক ছিল। মেয়েটির ঢেউ খেলানো চুল ছিল। মেয়েটির কানে দুল পরা ছিল। মেয়েটির লিপস্টিক পরা ছিল। "
 
3
  from diffusers import DiffusionPipeline
4
  from transformers import AutoTokenizer,AutoModel
5
  from diffusers.models import AutoencoderKL
6
+
7
  pipeline = DiffusionPipeline.from_pretrained(
8
  "CompVis/stable-diffusion-v1-4",
9
  text_encoder = AutoModel.from_pretrained("csebuetnlp/banglabert"),
10
  custom_pipeline="gr33nr1ng3r/Mukh-Oboyob"
11
  )
12
  pipeline.unet.load_attn_procs("gr33nr1ng3r/Mukh-Oboyob")
 
13
 
14
  def diffusion(text,num_inference_steps,guidance_scale):
15
  prompt = "মেয়েটির কালো চুল ছিল। মেয়েটির মুখে ভারী মেকাপ ছিল। মেয়েটির উঁচু গালের হাড় ছিল। মেয়েটির মুখ কিছুটা খোলা ছিল। মেয়েটির চেহারা ডিম্বাকৃতির। মেয়েটির চোখা নাক ছিল। মেয়েটির ঢেউ খেলানো চুল ছিল। মেয়েটির কানে দুল পরা ছিল। মেয়েটির লিপস্টিক পরা ছিল। "