um235 commited on
Commit
ae20708
·
verified ·
1 Parent(s): 5307c65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -45,7 +45,7 @@ def infer(
45
 
46
  pipe = None
47
  if (model_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora"):
48
- pipe=DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch_dtype,cross_attention_kwargs={"scale": 0.5})
49
  pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
50
  else:
51
  print("stable-diffusion-v1-5/stable-diffusion-v1-5 with lora")
@@ -99,7 +99,7 @@ with gr.Blocks(css=css) as demo:
99
  minimum=0,
100
  maximum=2,
101
  step=0.05,
102
- value=1, # Replace with defaults that work for your model
103
  )
104
 
105
 
 
45
 
46
  pipe = None
47
  if (model_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora"):
48
+ pipe=DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch_dtype,cross_attention_kwargs={"scale": lscale})
49
  pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
50
  else:
51
  print("stable-diffusion-v1-5/stable-diffusion-v1-5 with lora")
 
99
  minimum=0,
100
  maximum=2,
101
  step=0.05,
102
+ value=1,
103
  )
104
 
105