Update comfy/model_management.py

comfy/model_management.py

@@ -21,7 +21,7 @@ class CPUState(Enum):
 # Determine VRAM State
 vram_state = VRAMState.NORMAL_VRAM
 set_vram_to = VRAMState.NORMAL_VRAM
-cpu_state = CPUState.GPU
+cpu_state = CPUState.CPU
 
 total_vram = 0
 
@@ -115,7 +115,7 @@ def get_total_memory(dev=None, torch_total_too=False):
     else:
         return mem_total
 
-total_vram = get_total_memory(get_torch_device()) / (1024 * 1024)
+total_vram = 0
 total_ram = psutil.virtual_memory().total / (1024 * 1024)
 print("Total VRAM {:0.0f} MB, total RAM {:0.0f} MB".format(total_vram, total_ram))
 if not args.normalvram and not args.cpu:
@@ -209,8 +209,7 @@ elif args.novram:
     set_vram_to = VRAMState.NO_VRAM
 elif args.highvram or args.gpu_only:
     vram_state = VRAMState.HIGH_VRAM
-
-FORCE_FP32 = False
+FORCE_FP32 = True
 FORCE_FP16 = False
 if args.force_fp32:
     print("Forcing FP32, if this improves things please report it.")