fix cpu offload not working

app.py

@@ -197,7 +197,7 @@ def run_inference(config, config_history, progress=gr.Progress(track_tqdm=True))
             num_inference_steps = int(config["inference_steps"]), 
             guidance_scale = float(config["guidance_scale"])).images
 
-        if str(config["use_safetensors"]).lower() != 'false':
+        if str(config["cpu_offload"]).lower() != 'false':
             pipeline.enable_model_cpu_offload()
 
         if config['refiner'].lower() != 'none':
@@ -207,7 +207,7 @@ def run_inference(config, config_history, progress=gr.Progress(track_tqdm=True))
                 image=image,
             ).images
             
-            if str(config["use_safetensors"]).lower() != 'false':
+            if str(config["cpu_offload"]).lower() != 'false':
                 refiner.enable_model_cpu_offload()
 
         config_history.append(config.copy())