DepthEdgeDepth2StreetGenerator.ds_root = "/media/servml/SSD_1/ds" # файлы в каждой папке семпла: DepthEdgeDepth2StreetGenerator.edge_filename = "depth_drone_edge.png" DepthEdgeDepth2StreetGenerator.depth_filename = "depth_drone.png" DepthEdgeDepth2StreetGenerator.out_dir = "out_street_expC" DepthEdgeDepth2StreetGenerator.out_name = "street_gen.png" DepthEdgeDepth2StreetGenerator.overwrite = False # модели DepthEdgeDepth2StreetGenerator.base_model_id = "stabilityai/stable-diffusion-xl-base-1.0" # используем depth-controlnet и для edge, и для depth (works ok на старте) DepthEdgeDepth2StreetGenerator.controlnet_edge_id = "diffusers/controlnet-depth-sdxl-1.0" DepthEdgeDepth2StreetGenerator.controlnet_depth_id = "diffusers/controlnet-depth-sdxl-1.0" DepthEdgeDepth2StreetGenerator.device = "cuda" DepthEdgeDepth2StreetGenerator.torch_dtype = "fp16" DepthEdgeDepth2StreetGenerator.enable_xformers = False # генерация DepthEdgeDepth2StreetGenerator.size = 1024 DepthEdgeDepth2StreetGenerator.steps = 30 DepthEdgeDepth2StreetGenerator.guidance_scale = 5.0 DepthEdgeDepth2StreetGenerator.seed = 123 # ключевое для Exp C: DepthEdgeDepth2StreetGenerator.edge_scale = 1.0 DepthEdgeDepth2StreetGenerator.depth_scale = 0.5 DepthEdgeDepth2StreetGenerator.prompt = "realistic street-level photo, eye-level viewpoint, camera height 1.6m, wide-angle 24mm lens, street, sidewalks, buildings facades, road, natural daylight, high detail, sharp focus" DepthEdgeDepth2StreetGenerator.negative_prompt = "aerial view, drone view, top-down, bird's-eye view, satellite, map, isometric, miniature, tilt-shift, orthographic, lowres, blurry, warped perspective, distorted geometry" # если хочешь прогнать первые N сэмплов: DepthEdgeDepth2StreetGenerator.max_samples = 0