File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -399,12 +399,12 @@ def set_timesteps(
399399        if  self .config .use_karras_sigmas :
400400            sigmas  =  np .flip (sigmas ).copy ()
401401            sigmas  =  self ._convert_to_karras (in_sigmas = sigmas , num_inference_steps = num_inference_steps )
402-             timesteps  =  np .array ([self ._sigma_to_t (sigma , log_sigmas ) for  sigma  in  sigmas ]). round () 
402+             timesteps  =  np .array ([self ._sigma_to_t (sigma , log_sigmas ) for  sigma  in  sigmas ])
403403        elif  self .config .use_lu_lambdas :
404404            lambdas  =  np .flip (log_sigmas .copy ())
405405            lambdas  =  self ._convert_to_lu (in_lambdas = lambdas , num_inference_steps = num_inference_steps )
406406            sigmas  =  np .exp (lambdas )
407-             timesteps  =  np .array ([self ._sigma_to_t (sigma , log_sigmas ) for  sigma  in  sigmas ]). round () 
407+             timesteps  =  np .array ([self ._sigma_to_t (sigma , log_sigmas ) for  sigma  in  sigmas ])
408408        elif  self .config .use_exponential_sigmas :
409409            sigmas  =  np .flip (sigmas ).copy ()
410410            sigmas  =  self ._convert_to_exponential (in_sigmas = sigmas , num_inference_steps = num_inference_steps )
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments