File tree Expand file tree Collapse file tree 2 files changed +3
-0
lines changed Expand file tree Collapse file tree 2 files changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -499,6 +499,7 @@ def test_quantized_tensor_subclass_save_load_map_location(self):
499499 @unittest .skipIf (not torch .accelerator .is_available (), "Need GPU available" )
500500 def test_quantized_model_streaming (self ):
501501 device_module = torch .get_device_module (_DEVICE )
502+
502503 def reset_memory ():
503504 gc .collect ()
504505 device_module .empty_cache ()
@@ -1110,6 +1111,7 @@ def test_non_fqn_config_filter_fn_none(self):
11101111 @unittest .skipIf (not torch .accelerator .is_available (), "Need GPU available" )
11111112 def test_quantized_model_streaming_fqn_config (self ):
11121113 device_module = torch .get_device_module (_DEVICE )
1114+
11131115 def reset_memory ():
11141116 gc .collect ()
11151117 device_module .empty_cache ()
Original file line number Diff line number Diff line change 4343
4444_DEVICE = auto_detect_device ()
4545
46+
4647# Helper function to run a function twice
4748# and verify that the result is the same.
4849# Adds some verification to avoid side effects.
You can’t perform that action at this time.
0 commit comments