File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -356,6 +356,8 @@ jobs:
356356        config :
357357          - backend : " bitsandbytes" 
358358            test_location : " bnb" 
359+           - backend : " gguf" 
360+             test_location : " gguf" 
359361    runs-on :
360362      group : aws-g6e-xlarge-plus 
361363    container :
@@ -519,4 +521,4 @@ jobs:
519521#         if: always()
520522#         run: |
521523#           pip install slack_sdk tabulate
522- #           python utils/log_reports.py >> $GITHUB_STEP_SUMMARY
524+ #           python utils/log_reports.py >> $GITHUB_STEP_SUMMARY
Original file line number Diff line number Diff line change @@ -75,7 +75,7 @@ def test_gguf_linear_layers(self):
7575            if  isinstance (module , torch .nn .Linear ) and  hasattr (module .weight , "quant_type" ):
7676                assert  module .weight .dtype  ==  torch .uint8 
7777
78-     def  test_gguf_memory (self ):
78+     def  test_gguf_memory_usage (self ):
7979        quantization_config  =  GGUFQuantizationConfig (compute_dtype = self .torch_dtype )
8080
8181        model  =  FluxTransformer2DModel .from_single_file (
    
 
   
 
     
   
   
          
     
  
    
     
 
    
      
     
 
     
    You can’t perform that action at this time.
  
 
    
  
     
    
      
        
     
 
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments