Profakerr commited on
Commit
8d3fced
·
verified ·
1 Parent(s): 692438e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ lora_path = "OedoSoldier/detail-tweaker-lora"
8
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to("cuda")
9
 
10
  @spaces.GPU
11
- def generate_image(prompt, negative_prompt, num_inference_steps=30, guidance_scale=7.0,model="Real6.0",num_images=1, width, height):
12
 
13
  if model == "Real5.0":
14
  model_id = "SG161222/Realistic_Vision_V5.0_noVAE"
 
8
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to("cuda")
9
 
10
  @spaces.GPU
11
+ def generate_image(prompt, negative_prompt, num_inference_steps=30, guidance_scale=7.0,model="Real6.0",num_images=1, width=512, height=512):
12
 
13
  if model == "Real5.0":
14
  model_id = "SG161222/Realistic_Vision_V5.0_noVAE"