o h# @sdZddlZddlmZmZmZddlmZddlZddl m Z ddl m Z ddl Z ddZ e Zed kr9ed d ejd d ddejdddejdddZdddddddddddddddd d!dd"d#dd$d%dd&ZGd'd(d(ZdS))zV LOCAL GPU IMAGE GENERATOR - GPU OPTIMIZED ========================================== N)StableDiffusionXLPipelineStableDiffusionPipelineDPMSolverMultistepScheduler)Image)BytesIO)datetimecCstjs$tdtdtjtdtjjtdtddStdtdtjtdtjjtd tjd td tjd jd d ddS)Nu❌ CUDA not available!z PyTorch version: z CUDA compiled version: u 💡 Install PyTorch with CUDA:zS pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118cpuu✅ CUDA is available!z CUDA version: z GPU: rz VRAM: @z.1fz GBcuda) torchr is_availableprint __version__versionget_device_nameget_device_properties total_memoryrrJ/var/www/eduai.edurigo.com/generate_image/testing/local_image_generator.py detect_devices  rrzgGPU not detected! This application requires CUDA-enabled GPU. Please install PyTorch with CUDA support.z(stabilityai/stable-diffusion-xl-base-1.0fp16)r)repo_iddtypevariantrecommended_sizezrunwayml/stable-diffusion-v1-5)r)rrrz'SG161222/Realistic_Vision_V6.0_B1_noVAE)ri)sdxlsd15 realisticzjphotorealistic, 8k uhd, high quality, detailed, professional photography, sharp focus, dslr, soft lightingzEcartoon, anime, painting, sketch, low quality, blurry, ugly, deformed)positivenegativezU3d render, octane render, unreal engine 5, ray tracing, volumetric lighting, detailedz2d, cartoon, flat, low qualityzCcartoon style, disney pixar, animated, vibrant colors, professionalz%realistic photo, dark, ugly, deformedzPcomic book style, graphic novel, bold lines, vibrant colors, dynamic compositionzphoto, blurry, low qualityz?anime style, manga, studio quality, clean lines, vibrant colorsz!realistic, photo, western cartoonzUcinematic lighting, film scene, dramatic, professional cinematography, depth of fieldz"flat lighting, low quality, boringzIfantasy art, magical, epic, detailed, concept art, trending on artstationzmodern, realistic, low qualityz4cyberpunk, neon lights, futuristic, sci-fi, detailedzfantasy, medieval, low quality)r3d cartoonisticcomicanime cinematicfantasy cyberpunkc @seZdZdZddZdedefddZded efd d Zded eded e d e f ddZ ded eded e d e de f ddZ ddZ dS)LocalImageGeneratorz/Handles model loading and GPU image generation.cCsd|_d|_tddS)Nu-✅ Local Image Generator initialized on GPU.)pipeline current_modelr )selfrrr__init__cs zLocalImageGenerator.__init__ model_namereturnc Csh|j|kr|jrtd|ddS|jr)td|`ttjtjt |t d}td|dtdz\|dkrDt nt }|j |d |d d| d d |_t|jjj|j_|jt|_|jz |jtd Wn tytdYnw||_tdt|jjjWdSty}ztd|WYd}~dSd}~ww)z4Load model to GPU memory, reusing if already loaded.uℹ️ Model 'z' already loaded on GPU.Tu(🧹 Clearing previous model from GPU...ru📦 Loading model: z to GPUu+⏳ First load may take up to 60 seconds...rrr) torch_dtypeuse_safetensorsru*✅ xformers enabled for faster inference.u8ℹ️ xformers not available, using standard attention.u✅ Model loaded on: u❌ Model load failed: NF)r+r*r gccollectr r empty_cache synchronize MODEL_CONFIGSgetrrfrom_pretrainedr from_config schedulerconfigtoDEVICEenable_attention_slicing*enable_xformers_memory_efficient_attention Exceptionnextunet parametersdevice)r,r.cfgpipeline_classerrr load_modelhsN       zLocalImageGenerator.load_modelpromptcategorycCs.t|td}|d|d|dfS)z1Add style-specific positive and negative prompts.rz, r r!)CATEGORY_STYLESr7lower)r,rIrJstylerrrenhance_promptsz"LocalImageGenerator.enhance_promptwidthheightc Cs(||sdS|||\}}|dkrdnd}||d||}||d||}td|d|d|zsl   #