Spaces:
Runtime error
Runtime error
File size: 3,639 Bytes
09c7c70 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
def is_google_colab():
try:
import google.colab
return True
except:
return False
import os
def check_RealESRGAN():
try:
from RealESRGANv030.interface import realEsrgan
except:
os.system('cd RealESRGANv030 && python setup.py develop')
import argparse
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('-n', '--n',
type = int,
default = 0,
help = 'Generate image n times, (0=infinite)')
parser.add_argument('-mn', '--model_name',
type = str,
default = 'anything v3',
help = 'Models used to generate anime images.'
)
parser.add_argument('-wd', '--words',
type = str,
default = "1girl, brown hair, green eyes, colorful, autumn, \
cumulonimbus clouds, lighting, blue sky, falling leaves, garden",
help = 'Text (prompt keys) used to generate anime image'
)
parser.add_argument('-nwd', '--neg_words',
type = str,
default = "lowers, bad anatomy, bad hands, text, error, \
missing fingers, extra digit, fewer digits, cropped, worst quality, \
low quality, normal quality, jpeg artifacts, signature, watermark, \
username, blurry, artist name, bad feet",
help = 'Negative prompt keys'
)
parser.add_argument('-img', '--image',
type = str,
default = '',
help = 'Input image used to generate anime figure.')
parser.add_argument('-w', '--width',
type = int,
default = 512,
help = 'Image width before resolution scale',
)
parser.add_argument('-h_', '--height',
type = int,
default = 512,
help = 'Image height before resolution scale')
parser.add_argument('-g', '--guidance',
type = float,
default = 7.5,
help = 'Guidance for anime image generation.')
parser.add_argument('-st', '--strength',
type = float,
default = 0.5,
help = 'Transform strength.')
parser.add_argument('-gs', '--gen_steps',
type = int,
default = 20,
help = 'Steps for anime image generation process.')
parser.add_argument('-seed',
type = int,
default = 0,
help = 'Random seed, 0 means random')
parser.add_argument('-s', '--scale',
type = int,
default = 4,
help = 'Resolution scale (> 1)')
parser.add_argument('-t', '--tile',
type = int,
default = 0,
help = 'Tile for resolution up scale, 0 = no tile')
parser.add_argument('-e', '--extension',
type = str,
default = 'auto',
help = 'extension of output image')
parser.add_argument('-o', '--out_dir',
type = str,
default = 'imgs',
help = 'Directory used to save images.')
return parser.parse_args() |