* Add threshold and perlin noise options for Karras samplers.

This commit is contained in:
Peter Baylies
2022-09-02 13:39:26 -04:00
parent 3ee82d8a3b
commit b6cf8b9052
4 changed files with 71 additions and 4 deletions

View File

@ -9,6 +9,7 @@ import sys
import copy
import warnings
import time
sys.path.insert(0, '.')
from ldm.dream.devices import choose_torch_device
import ldm.dream.readline
from ldm.dream.pngwriter import PngWriter, PromptFormatter
@ -546,6 +547,18 @@ def create_cmd_parser():
action='store_true',
help='shows how the prompt is split into tokens'
)
parser.add_argument(
'--threshold',
default=0.0,
type=float,
help='Add threshold value aka perform clipping.',
)
parser.add_argument(
'--perlin',
default=0.0,
type=float,
help='Add perlin noise.',
)
return parser