Commit cd1aeb01 authored by kurumuz's avatar kurumuz

hack

parent 2ef44e5d
......@@ -146,6 +146,7 @@ def init_config_model():
config.module_path = os.getenv('MODULE_PATH', None)
config.prior_path = os.getenv('PRIOR_PATH', None)
config.default_config = os.getenv('DEFAULT_CONFIG', None)
config.quality_hack = os.getenv('QUALITY_HACK', "0")
# Misc settings
config.model_alias = os.getenv('MODEL_ALIAS')
......
......@@ -358,7 +358,11 @@ class StableDiffusionModel(nn.Module):
uc = [request.uc] * request.n_samples
uc = prompt_mixing(self.model, uc[0], request.n_samples)
else:
uc = self.model.get_learned_conditioning(request.n_samples * [""])
if self.config.quality_hack == "1":
uc = ["Tags: lowres"] * request.n_samples
uc = prompt_mixing(self.model, uc[0], request.n_samples)
else:
uc = self.model.get_learned_conditioning(request.n_samples * [""])
shape = [
request.latent_channels,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment