File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed 
src/diffusers/models/transformers Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 3232    get_1d_sincos_pos_embed_from_grid ,
3333)
3434from  ..modeling_outputs  import  Transformer2DModelOutput 
35- from  ..modeling_utils  import  ModelMixin 
35+ from  ..modeling_utils  import  ModelMixin ,  get_parameter_dtype 
3636from  ..normalization  import  FP32LayerNorm 
3737
3838
@@ -198,7 +198,7 @@ def forward(
198198    ):
199199        timestep  =  self .timesteps_proj (timestep )
200200
201-         time_embedder_dtype  =  next ( iter ( self .time_embedder . parameters ())). dtype 
201+         time_embedder_dtype  =  get_parameter_dtype ( self .time_embedder ) 
202202        if  timestep .dtype  !=  time_embedder_dtype  and  time_embedder_dtype  !=  torch .int8 :
203203            timestep  =  timestep .to (time_embedder_dtype )
204204        temb  =  self .time_embedder (timestep ).type_as (encoder_hidden_states )
 
 
   
 
     
   
   
          
    
    
     
    
      
     
     
    You can’t perform that action at this time.
  
 
    
  
    
      
        
     
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments