Hugging Face's logo

Spaces:
Aatricks
/
LightDiffusion-Next
Running on Zero

App Files Community
LightDiffusion-Next
File size: 4,193 Bytes 
d9a2e19
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1d117d0
 
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
 
import os
import re

files_ordered = [
    "./modules/Utilities/util.py",
    "./modules/sample/sampling_util.py",
    "./modules/Device/Device.py",
    "./modules/cond/cond_util.py",
    "./modules/cond/cond.py",
    "./modules/sample/ksampler_util.py",
    "./modules/cond/cast.py",
    "./modules/Attention/AttentionMethods.py",
    "./modules/AutoEncoders/taesd.py",
    "./modules/cond/cond.py",
    "./modules/cond/Activation.py",
    "./modules/Attention/Attention.py",
    "./modules/sample/samplers.py",
    "./modules/sample/CFG.py",
    "./modules/NeuralNetwork/transformer.py",
    "./modules/sample/sampling.py",
    "./modules/clip/CLIPTextModel.py",
    "./modules/AutoEncoders/ResBlock.py",
    "./modules/AutoDetailer/mask_util.py",
    "./modules/NeuralNetwork/unet.py",
    "./modules/SD15/SDClip.py",
    "./modules/SD15/SDToken.py",
    "./modules/UltimateSDUpscale/USDU_util.py",
    "./modules/StableFast/SF_util.py",
    "./modules/Utilities/Latent.py",
    "./modules/AutoDetailer/SEGS.py",
    "./modules/AutoDetailer/tensor_util.py",
    "./modules/AutoDetailer/AD_util.py",
    "./modules/clip/FluxClip.py",
    "./modules/Model/ModelPatcher.py",
    "./modules/Model/ModelBase.py",
    "./modules/UltimateSDUpscale/image_util.py",
    "./modules/UltimateSDUpscale/RDRB.py",
    "./modules/StableFast/ModuleFactory.py",
    "./modules/AutoDetailer/bbox.py",
    "./modules/AutoEncoders/VariationalAE.py",
    "./modules/clip/Clip.py",
    "./modules/Model/LoRas.py",
    "./modules/BlackForest/Flux.py",
    "./modules/UltimateSDUpscale/USDU_upscaler.py",
    "./modules/StableFast/ModuleTracing.py",
    "./modules/hidiffusion/utils.py",
    "./modules/FileManaging/Downloader.py",
    "./modules/AutoDetailer/SAM.py",
    "./modules/AutoDetailer/ADetailer.py",
    "./modules/Quantize/Quantizer.py",
    "./modules/FileManaging/Loader.py",
    "./modules/SD15/SD15.py",
    "./modules/UltimateSDUpscale/UltimateSDUpscale.py",
    "./modules/StableFast/StableFast.py",
    "./modules/hidiffusion/msw_msa_attention.py",
    "./modules/FileManaging/ImageSaver.py",
    "./modules/Utilities/Enhancer.py",
    "./modules/Utilities/upscale.py",
    "./modules/user/pipeline.py",
]

def get_file_patterns():
    patterns = []
    seen = set()
    for path in files_ordered:
        filename = os.path.basename(path)
        name = os.path.splitext(filename)[0]
        if name not in seen:
            # Pattern 1: matches module name when not in brackets or after a dot
            pattern1 = rf'(?<![a-zA-Z0-9_\.])({name}\.)(?![)\]])'
            # Pattern 2: matches module name inside brackets while preserving them  
            pattern2 = rf'(\[|\()({name}\.)([^\]\)]+?)(\]|\))'
            pattern3 = 'cond_util\.'
            patterns.extend([
                (pattern1, ''),  # Remove module name and dot outside brackets
                (pattern2, r'\1\3\4'),  # Keep brackets, remove only module name
                (pattern3, '')
            ])
            seen.add(name)
    return patterns

def remove_file_names(line):
    patterns = get_file_patterns()
    result = line
    for pattern, replacement in patterns:
        result = re.sub(pattern, replacement, result)
    return result

try:
    with open("./compiled.py", "w") as output_file:
        for file_path in files_ordered:
            try:
                with open(file_path, "r") as input_file:
                    for line in input_file:
                        if not line.lstrip().startswith("from modules."):
                            # Apply the file name removal before writing
                            modified_line = remove_file_names(line)
                            output_file.write(modified_line)
                    output_file.write("\n\n")
                print(f"Processed: {file_path}")
            except FileNotFoundError:
                print(f"Error: Could not find file {file_path}")
            except Exception as e:
                print(f"Error processing {file_path}: {str(e)}")
except Exception as e:
    print(f"Error creating compiled.py: {str(e)}")