Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
vars.py 3.19 KB
Newer Older
April Hu's avatar
Add flux1 demo for depth and canny  
April Hu committed
1
2
3
4
5
6
7
8
9
10
11
12 
STYLES = {
    "None": "{prompt}",
    "Cinematic": "cinematic still {prompt}. emotional, harmonious, vignette, highly detailed, high budget, bokeh, cinemascope, moody, epic, gorgeous, film grain, grainy",
    "3D Model": "professional 3d model {prompt}. octane render, highly detailed, volumetric, dramatic lighting",
    "Anime": "anime artwork {prompt}. anime style, key visual, vibrant, studio anime,  highly detailed",
    "Digital Art": "concept art {prompt}. digital artwork, illustrative, painterly, matte painting, highly detailed",
    "Photographic": "cinematic photo {prompt}. 35mm photograph, film, bokeh, professional, 4k, highly detailed",
    "Pixel art": "pixel-art {prompt}. low-res, blocky, pixel art style, 8-bit graphics",
    "Fantasy art": "ethereal fantasy concept art of  {prompt}. magnificent, celestial, ethereal, painterly, epic, majestic, magical, fantasy art, cover art, dreamy",
    "Neonpunk": "neonpunk style {prompt}. cyberpunk, vaporwave, neon, vibes, vibrant, stunningly beautiful, crisp, detailed, sleek, ultramodern, magenta highlights, dark purple shadows, high contrast, cinematic, ultra detailed, intricate, professional",
    "Manga": "manga style {prompt}. vibrant, high-energy, detailed, iconic, Japanese comic style",
}
muyangli's avatar
[major] add flux.1-redux; update the flux.1-tools demos  
muyangli committed
13 
DEFAULT_STYLE_NAME = "None"
April Hu's avatar
Add flux1 demo for depth and canny  
April Hu committed
14
15
16
17
18
19
20
21 
STYLE_NAMES = list(STYLES.keys())

MAX_SEED = 1000000000
DEFAULT_INFERENCE_STEP_CANNY = 50
DEFAULT_GUIDANCE_CANNY = 30.0

DEFAULT_INFERENCE_STEP_DEPTH = 30
DEFAULT_GUIDANCE_DEPTH = 10.0
muyangli's avatar
[major] add flux.1-redux; update the flux.1-tools demos  
muyangli committed
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 

HEIGHT = 1024
WIDTH = 1024

EXAMPLES = {
    "canny": [
        [
            "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/robot.png",
            "A robot made of exotic candies and chocolates of different kinds. The background is filled with confetti and celebratory gifts.",
            DEFAULT_STYLE_NAME,
            STYLES[DEFAULT_STYLE_NAME],
            50,
            30,
            0,
        ],
        [
            "https://huggingface.co/mit-han-lab/svdq-int4-flux.1-fill-dev/resolve/main/example.png",
            "A wooden basked of several individual cartons of strawberries.",
            DEFAULT_STYLE_NAME,
            STYLES[DEFAULT_STYLE_NAME],
            50,
            30,
            1,
        ],
    ],
    "depth": [
        [
            "https://huggingface.co/mit-han-lab/svdq-int4-flux.1-canny-dev/resolve/main/logo_example.png",
            "A logo of 'MIT HAN Lab'.",
            DEFAULT_STYLE_NAME,
            STYLES[DEFAULT_STYLE_NAME],
            30,
            10,
            2,
        ],
        [
            "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/robot.png",
            "A robot made of exotic candies and chocolates of different kinds. The background is filled with confetti and celebratory gifts.",
            DEFAULT_STYLE_NAME,
            STYLES[DEFAULT_STYLE_NAME],
            30,
            10,
            0,
        ],
        [
            "https://huggingface.co/mit-han-lab/svdq-int4-flux.1-fill-dev/resolve/main/example.png",
            "A wooden basked of several individual cartons of strawberries.",
            DEFAULT_STYLE_NAME,
            STYLES[DEFAULT_STYLE_NAME],
            30,
            10,
            1,
        ],
    ],
}