Spaces:

ktrndy
/

diffusion-image-gen

Sleeping

ktrndy commited on Feb 9

Commit

1006877

verified ·

1 Parent(s): bb0fa7b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ MAX_IMAGE_SIZE = 1024
 def get_lora_sd_pipeline(
-    ckpt_dir='./model_output',
     base_model_name_or_path=model_id_default,
     dtype=torch_dtype,
     device=device
@@ -81,7 +81,7 @@ def infer(
     seed=42,
     guidance_scale=7.0,
     lora_scale=1.0,
-    num_inference_steps=20,
     progress=gr.Progress(track_tqdm=True),
 ):
     generator = torch.Generator(device).manual_seed(seed)
@@ -99,7 +99,6 @@ def infer(
         height=height,
         generator=generator,
     ).images[0]
-    print(device, torch_dtype)
     return image

 def get_lora_sd_pipeline(
+    ckpt_dir='./output',
     base_model_name_or_path=model_id_default,
     dtype=torch_dtype,
     device=device
     seed=42,
     guidance_scale=7.0,
     lora_scale=1.0,
+    num_inference_steps=30,
     progress=gr.Progress(track_tqdm=True),
 ):
     generator = torch.Generator(device).manual_seed(seed)
         height=height,
         generator=generator,
     ).images[0]
     return image