diff --git a/modules/shared.py b/modules/shared.py index 2a366248d..eebbfef20 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -89,6 +89,8 @@ devices.device = devices.get_optimal_device() mem_stat = memory_stats() cpu_memory = round(mem_stat['ram']['total'] if "ram" in mem_stat else 0) gpu_memory = round(mem_stat['gpu']['total'] if "gpu" in mem_stat else 0) +if gpu_memory == 0: + gpu_memory = cpu_memory native = backend == Backend.DIFFUSERS if not files_cache.do_cache_folders: log.warning('File cache disabled: ') diff --git a/modules/shared_defaults.py b/modules/shared_defaults.py index bc7f8ce11..8ada27a0c 100644 --- a/modules/shared_defaults.py +++ b/modules/shared_defaults.py @@ -10,7 +10,7 @@ def get_default_modes(cmd_opts, mem_stat): default_diffusers_offload_never = '' gpu_memory = round(mem_stat['gpu']['total'] if "gpu" in mem_stat else 0) if not (cmd_opts.lowvram or cmd_opts.medvram): - if "gpu" in mem_stat: + if "gpu" in mem_stat and gpu_memory != 0: if gpu_memory <= 4: cmd_opts.lowvram = True default_offload_mode = "sequential"