o ?\iW' @sjdZddlZddlZddlZddlmZddlmZddlmZm Z m Z m Z ddl Z ddl mZmZdde jd ed iZd d d ddd ddd ddd ddd ddd ddd ddd dZdad4deddfd d!Zdefd"d#Zd$ed%ed&ede eeffd'd(Zd)ed*ede eeffd+d,ZGd-d.d.Zdefd/d0Zd1e de e eeffd2d3Zd3d.gZdS)5z} LOCAL GPU IMAGE GENERATOR (Optimized for SD15) Lazy GPU init + strong diagnostics so systemd/env issues don't crash import. N)BytesIO)datetime)OptionalDictListTuple)StableDiffusionPipelineDPMSolverMultistepSchedulersd15zrunwayml/stable-diffusion-v1-5@)repo_iddtype size_multiplepipeline_classzPphotorealistic, 8k uhd, professional photography, detailed lighting, ultra sharpz8cartoon, painting, sketch, low quality, blurry, deformed)positivenegativez;anime style, studio quality, vibrant colors, smooth shadingz'realistic photo, ugly, deformed, blurryz43d render, octane render, unreal engine, ray tracingz2d, cartoon, low qualityz1cartoon, colorful, animated style, clean outlineszrealistic, photo, blurryz1comic book style, inking, halftone, bold outlineszphoto, realistic, blurryzEcinematic lighting, professional composition, dramatic depth of fieldzflat, low quality, dullzBfantasy concept art, magic, epic, detailed, trending on artstationzmodern, realistic, low qualityz>cyberpunk, neon lights, futuristic, sci-fi, detailed cityscapezfantasy, medieval, low quality) realisticanime3d cartoonisticcomic cinematicfantasy cyberpunkprefixreturnc Cszkt|dtjt|dttjddt|dtjt|dtj dzddl }|j d g|j d d d }t|d |WWdSt yk}zt|d|WYd}~WdSd}~wwt y}zt|d|WYd}~dSd}~ww)z%Print detailed CUDA diagnostics once.ztorch.cuda.is_available(): ztorch.version.cuda: cudaNztorch.__version__: zCUDA_VISIBLE_DEVICES: CUDA_VISIBLE_DEVICESrz nvidia-smiT)stderrtexttimeoutznvidia-smi OK: z$nvidia-smi not available or failed: zCUDA diagnostic failed: )printtorchr is_availablegetattrversion __version__osenvironget subprocess check_outputSTDOUT Exception)rr,smier2D/var/www/eduai.edurigo.com/roleplay/testing/local_image_generator.py_diagnose_cudaBs"" r4cCs,tjdddkr dStjrdSdS)zsd   $"u ,