Update app.py

app.py

@@ -167,7 +167,7 @@ def infer(ref_style_file, style_description, caption, progress):
 
         if low_vram:
             # The sampling process uses more vram, so we offload everything except two modules to the cpu.
-            models_to(models_rbm, device="cpu", excepts=["generator", "previewer"])
+            models_to(models_rbm, device="cpu")
 
         progress(0.4, "Starting Stage C reverse process")
         # Stage C reverse process.
@@ -283,7 +283,7 @@ def infer_compo(style_description, ref_style_file, caption, ref_sub_file, progre
         unconditions_b = core_b.get_conditions(batch, models_b, extras_b, is_eval=True, is_unconditional=True)
 
         if low_vram:
-            models_to(models_rbm, device="cpu", excepts=["generator", "previewer"])
+            models_to(models_rbm, device="cpu")
             models_to(sam_model, device="cpu")
             models_to(sam_model.sam, device="cpu")