let me cook

backend/loader.py

@@ -379,7 +379,7 @@ def load_huggingface_component(guess, component_name, lib_name, cls_name, repo_p
 
             override_dtype = backend.args.dynamic_args.forge_unet_storage_dtype
             if override_dtype is torch.int8:
-                override_dtype = None
+                override_dtype = torch.bfloat16 if memory_management.should_use_bf16(load_device) else None
                 try_int8 = True
 
             if guess.nunchaku:

backend/operations.py

@@ -404,7 +404,7 @@ class ForgeOperationsInt8(ForgeOperations):
             self.compute_dtype = torch.bfloat16
             self.lora_patches = []  # List of (down_scaled, up, start, size) set by INT8ModelPatcher
 
-            self.parameters_manual_cast = True
+            self.parameters_manual_cast = current_dtype != self.compute_dtype
 
         def reset_parameters(self):
             return None