File tree Expand file tree Collapse file tree 1 file changed +1
-21
lines changed 
tests/models/transformers Expand file tree Collapse file tree 1 file changed +1
-21
lines changed Original file line number Diff line number Diff line change 1919
2020from  diffusers  import  LTXVideoTransformer3DModel 
2121from  diffusers .utils .testing_utils  import  (
22-     enable_full_determinism ,
23-     is_torch_compile ,
24-     require_torch_2 ,
25-     require_torch_gpu ,
26-     slow ,
27-     torch_device ,
28- )
22+ from  diffusers .utils .testing_utils  import  enable_full_determinism , torch_device 
2923
3024from  ..test_modeling_common  import  ModelTesterMixin , TorchCompileTesterMixin 
3125
@@ -87,20 +81,6 @@ def prepare_init_args_and_inputs_for_common(self):
8781
8882    def  test_gradient_checkpointing_is_applied (self ):
8983        expected_set  =  {"LTXVideoTransformer3DModel" }
90-         super ().test_gradient_checkpointing_is_applied (expected_set = expected_set )
91- 
92-     @require_torch_gpu  
93-     @require_torch_2  
94-     @is_torch_compile  
95-     @slow  
96-     def  test_torch_compile_recompilation_and_graph_break (self ):
9784        torch ._dynamo .reset ()
98-         init_dict , inputs_dict  =  self .prepare_init_args_and_inputs_for_common ()
99- 
100-         model  =  self .model_class (** init_dict ).to (torch_device )
101-         model .eval ()
102-         model  =  torch .compile (model , fullgraph = True )
103- 
104-         with  torch ._dynamo .config .patch (error_on_recompile = True ), torch .no_grad ():
10585            _  =  model (** inputs_dict )
10686            _  =  model (** inputs_dict )
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments