Browse Source
        
      
      Fix config issue to support Triton
      
        Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
      
      
        main
      
      
     
    
    
    
	
		
			
				 1 changed files with 
3 additions and 
1 deletions
			 
			
		 
		
			
				- 
					
					
					 
					timm_image.py
				
 
			
		
		
			
			
			
			
			
			
				
				
					
						
							
								
									
	
		
			
				| 
					
					
						
							
						
					
					
				 | 
				@ -120,7 +120,9 @@ class TimmImage(NNOperator): | 
			
		
		
	
		
			
				 | 
				 | 
				
 | 
				 | 
				 | 
				
 | 
			
		
		
	
		
			
				 | 
				 | 
				    @property | 
				 | 
				 | 
				    @property | 
			
		
		
	
		
			
				 | 
				 | 
				    def config(self): | 
				 | 
				 | 
				    def config(self): | 
			
		
		
	
		
			
				 | 
				 | 
				        return resolve_data_config({}, model=self._model) | 
				 | 
				 | 
				 | 
			
		
		
	
		
			
				 | 
				 | 
				 | 
				 | 
				 | 
				        m = create_model(self.model_name, pretrained=False) | 
			
		
		
	
		
			
				 | 
				 | 
				 | 
				 | 
				 | 
				        config = resolve_data_config({}, model=m) | 
			
		
		
	
		
			
				 | 
				 | 
				 | 
				 | 
				 | 
				        return config | 
			
		
		
	
		
			
				 | 
				 | 
				
 | 
				 | 
				 | 
				
 | 
			
		
		
	
		
			
				 | 
				 | 
				    @arg(1, to_image_color('RGB')) | 
				 | 
				 | 
				    @arg(1, to_image_color('RGB')) | 
			
		
		
	
		
			
				 | 
				 | 
				    def convert_img(self, img: towhee._types.Image): | 
				 | 
				 | 
				    def convert_img(self, img: towhee._types.Image): | 
			
		
		
	
	
		
			
				| 
					
						
							
						
					
					
					
				 | 
				
  |