diff --git a/modules/devices.py b/modules/devices.py
index 4e0d63722..fbcf0146a 100644
--- a/modules/devices.py
+++ b/modules/devices.py
@@ -197,7 +197,7 @@ def get_optimal_device():
     return torch.device(get_optimal_device_name())
 
 
-def torch_gc(force:bool=False, fast:bool=False, reason:str=None):
+def torch_gc(force: bool = False, fast: bool = False, reason: str | None = None):
     def get_stats():
         mem_dict = memstats.memory_stats()
         gpu_dict = mem_dict.get('gpu', {})