Hugging Face's logo

Spaces:
ozgurkara
/
RAVE
Running on A10G

App Files Community
1
RAVE
File size: 1,554 Bytes 
eb9a9b4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2536180
eb9a9b4
 
 
 
2536180
eb9a9b4
2536180
 
eb9a9b4
 
 
 
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
 

import os
from datetime import datetime 

date_time = datetime.now().strftime("%m-%d-%Y")

CWD = os.getcwd()
MP4_PATH = os.path.join(CWD, 'data', 'mp4_videos')
OUTPUT_PATH = os.path.join(CWD, 'results', date_time) 

GENERATED_DATA_PATH = os.path.join(CWD, 'generated', 'data')
PREPROCESSOR_DICT = {
    'lineart_realistic': "lllyasviel/control_v11p_sd15_lineart",
    'lineart_coarse': "lllyasviel/control_v11p_sd15_lineart",
    'lineart_standard': "lllyasviel/control_v11p_sd15_lineart",
    'lineart_anime': "lllyasviel/control_v11p_sd15s2_lineart_anime",
    'lineart_anime_denoise': "lllyasviel/control_v11p_sd15s2_lineart_anime",
    'softedge_hed': 'lllyasviel/control_v11p_sd15_softedge',
    'softedge_hedsafe': 'lllyasviel/control_v11p_sd15_softedge',
    'softedge_pidinet': 'lllyasviel/control_v11p_sd15_softedge',
    'softedge_pidsafe': 'lllyasviel/control_v11p_sd15_softedge',
    'canny': 'lllyasviel/control_v11p_sd15_canny',
    'depth_leres': 'lllyasviel/control_v11f1p_sd15_depth',
    'depth_leres++': 'lllyasviel/control_v11f1p_sd15_depth',
    'depth_midas': 'lllyasviel/control_v11f1p_sd15_depth',
    'depth_zoe': 'lllyasviel/control_v11f1p_sd15_depth',
}

MODEL_IDS = {
    # 'Realistic Vision V5.1': '130072',
    'Realistic Vision V6.0' : '245598',
    'MajicMIXRealisticV7' : '176425',
    'DreamShaper' : '128713',
    'EpicPhotoGasm' : '223670',
    # 'DivineEleganceMix (Anime)': '238656',
    'GhostMix (Anime)': '76907',
    # 'CetusMix (Anime)': '105924',
    # 'Counterfeit (Anime)': '57618',
    'SD 1.5': 'None'
}