@@ -112,6 +112,7 @@ def testExportT5Model(self):
112112 new_blob_name = new_blob_name [0 ] + "-pass.mlir"
113113 turbine_tank .changeBlobName (blob_name , new_blob_name )
114114 del current_args
115+ del turbine
115116
116117 def testExportClipVitLarge14 (self ):
117118 current_args = copy .deepcopy (default_arguments )
@@ -152,6 +153,8 @@ def testExportClipVitLarge14(self):
152153 if platform .system () != "Windows" :
153154 os .remove (current_args ["external_weight_path" ])
154155 os .remove (current_args ["vmfb_path" ])
156+ del current_args
157+ del turbine
155158
156159 def testExportClipModel (self ):
157160 current_args = copy .deepcopy (default_arguments )
@@ -190,7 +193,10 @@ def testExportClipModel(self):
190193 if platform .system () != "Windows" :
191194 os .remove (current_args ["external_weight_path" ])
192195 os .remove (current_args ["vmfb_path" ])
196+ del current_args
197+ del turbine
193198
199+ @unittest .expectedFailure
194200 def testExportUnetModel (self ):
195201 current_args = copy .deepcopy (default_arguments )
196202 blob_name = unet .export_unet_model (
@@ -301,6 +307,7 @@ def testExportVaeModelDecode(self):
301307 new_blob_name = blob_name .split ("." )
302308 new_blob_name = new_blob_name [0 ] + "-pass.mlir"
303309 turbine_tank .changeBlobName (blob_name , new_blob_name )
310+ del current_args
304311 del torch_output
305312 del turbine
306313 os .remove ("stable_diffusion_v1_4_vae.safetensors" )
@@ -352,6 +359,8 @@ def testExportVaeModelEncode(self):
352359 turbine_tank .changeBlobName (blob_name , new_blob_name )
353360 os .remove ("stable_diffusion_v1_4_vae.safetensors" )
354361 os .remove ("stable_diffusion_v1_4_vae.vmfb" )
362+ del current_args
363+ del turbine
355364
356365 @unittest .expectedFailure
357366 def testExportPNDMScheduler (self ):
@@ -405,6 +414,7 @@ def testExportPNDMScheduler(self):
405414 turbine_tank .changeBlobName (blob_name , new_blob_name )
406415 os .remove ("stable_diffusion_v1_4_scheduler.safetensors" )
407416 os .remove ("stable_diffusion_v1_4_scheduler.vmfb" )
417+ del current_args
408418 del torch_output
409419 del turbine
410420
0 commit comments