Skip to content

Commit e109b5e

Browse files
remove limit
1 parent e71185d commit e109b5e

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

web/infer.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -59,11 +59,7 @@ def get_lora_paths():
5959
if not os.path.exists(lora_dir):
6060
os.makedirs(lora_dir, exist_ok=True)
6161
return ["None"]
62-
checkpoint_dirs = [
63-
d
64-
for d in os.listdir(lora_dir)
65-
if os.path.isdir(os.path.join(lora_dir, d)) and d.startswith("checkpoint")
66-
]
62+
checkpoint_dirs = [d for d in os.listdir(lora_dir) if os.path.isdir(os.path.join(lora_dir, d))]
6763

6864
if not checkpoint_dirs:
6965
return ["None"]
@@ -206,7 +202,7 @@ def refresh_lora_paths():
206202
minimum=1,
207203
maximum=50,
208204
step=1,
209-
value=1,
205+
value=50,
210206
)
211207
guidance_scale = gr.Slider(
212208
label="Guidance Scale",

0 commit comments

Comments
 (0)