4.在webui-user.bat加入下面的參數
set COMMANDLINE_ARGS=---medvram -precision full --no-half --skip-torch-cuda-test #AMD
set COMMANDLINE_ARGS=--xformers --medvram --precision full --no-half #NVIDIA 低階卡
set COMMANDLINE_ARGS=--xformers #NVIDIA 高階卡
如果 cuda 持續 out of memory 也可以試試加入 --opt-split-attention 或加入下面一行
set PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.6,max_split_size_mb:256 <-數字越小越安全
如果要使用ControlNet要移除 --medvram 不然會有錯誤如下:
Expected all tensors to be on the same device, but found at least two devices
目前設定:
set COMMANDLINE_ARGS=--xformers --opt-split-attention
set PYTORCH_CUDA_ALLOC_CONF=garbage_collection_threshold:0.6,max_split_size_mb:128
沒有留言:
張貼留言