Rename app.py--no-half-vae to app.py
Browse files- app.py--no-half-vae → app.py +25 -0
app.py--no-half-vae → app.py
RENAMED
@@ -26,6 +26,31 @@ def error_str(error, title="Error"):
|
|
26 |
return f"""#### {title}
|
27 |
{error}""" if error else ""
|
28 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
|
30 |
generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
|
31 |
prompt = f"{prefix} {prompt}" if auto_prefix else prompt
|
|
|
26 |
return f"""#### {title}
|
27 |
{error}""" if error else ""
|
28 |
|
29 |
+
|
30 |
+
def _parse_args(prompt, generator):
|
31 |
+
parser = argparse.ArgumentParser(
|
32 |
+
description="making it work."
|
33 |
+
)
|
34 |
+
parser.add_argument(
|
35 |
+
"--no-half-vae", help="no half vae"
|
36 |
+
)
|
37 |
+
|
38 |
+
cmdline_args = parser.parse_args()
|
39 |
+
command = cmdline_args.command
|
40 |
+
conf_file = cmdline_args.conf_file
|
41 |
+
conf_args = Arguments(conf_file)
|
42 |
+
opt = conf_args.readArguments()
|
43 |
+
|
44 |
+
if cmdline_args.config_overrides:
|
45 |
+
for config_override in cmdline_args.config_overrides.split(";"):
|
46 |
+
config_override = config_override.strip()
|
47 |
+
if config_override:
|
48 |
+
var_val = config_override.split("=")
|
49 |
+
assert (
|
50 |
+
len(var_val) == 2
|
51 |
+
), f"Config override '{var_val}' does not have the form 'VAR=val'"
|
52 |
+
conf_args.add_opt(opt, var_val[0], var_val[1], force_override=True)
|
53 |
+
|
54 |
def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
|
55 |
generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
|
56 |
prompt = f"{prefix} {prompt}" if auto_prefix else prompt
|