auto uninstall pynvml

Signed-off-by: Vladimir Mandic <mandic00@live.com>

CHANGELOG.md

@@ -43,6 +43,7 @@
   - **reference models** additional example images, thanks @liutyi  
   - **video** support for configurable multi-stage models such as WAN-2.2-14B  
   - **video** new LTX model selection  
+  - replace `pynvml` with `nvidia-ml-py` for gpu monitoring  
 - **Fixes**
   - framepack: add explicit hf-login before framepack load  
   - benchmark: remove forced sampler from system info benchmark  

installer.py

@@ -877,6 +877,7 @@ def install_torch_addons():
         install('pillow-jxl-plugin==1.3.4', 'pillow-jxl-plugin')
     if not args.experimental:
         uninstall('wandb', quiet=True)
+        uninstall('pynvml', quiet=True)
     ts('addons', t_start)
 
 

modules/processing_diffusers.py

@@ -80,13 +80,13 @@ def process_pre(p: processing.StableDiffusionProcessing):
         pag.apply(p)
         cfgzero.apply(p)
         linfusion.apply(shared.sd_model)
+        cachedit.apply_cache_dit(shared.sd_model)
 
         # apply-only
         sd_hijack_freeu.apply_freeu(p)
         transformer_cache.set_cache()
         para_attention.apply_first_block_cache()
         teacache.apply_teacache(p)
-        cachedit.apply_cache_dit(shared.sd_model)
     except Exception as e:
         shared.log.error(f'Processing apply: {e}')
         errors.display(e, 'apply')