Skip to content

Commit 9e599c6

Browse files
db3000lstein
authored andcommitted
Only output facetool parameters if enhancing faces
1 parent b953f82 commit 9e599c6

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

ldm/invoke/args.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -244,13 +244,12 @@ def dream_prompt_str(self,**kwargs):
244244
else:
245245
switches.append(f'-A {a["sampler_name"]}')
246246

247-
# facetool-specific parameters
248-
if a['facetool']:
249-
switches.append(f'-ft {a["facetool"]}')
247+
# facetool-specific parameters, only print if running facetool
250248
if a['facetool_strength']:
251249
switches.append(f'-G {a["facetool_strength"]}')
252-
if a['codeformer_fidelity']:
253-
switches.append(f'-cf {a["codeformer_fidelity"]}')
250+
switches.append(f'-ft {a["facetool"]}')
251+
if a["facetool"] == "codeformer":
252+
switches.append(f'-cf {a["codeformer_fidelity"]}')
254253

255254
if a['outcrop']:
256255
switches.append(f'-c {" ".join([str(u) for u in a["outcrop"]])}')

0 commit comments

Comments
 (0)