Update autoregressive/models/gpt_t2i.py

autoregressive/models/gpt_t2i.py

@@ -434,9 +434,9 @@ class Transformer(nn.Module):
                 if condition is not None:
                     condition_embeddings = self.condition_mlp(condition,train=self.training)#.to(torch.bfloat16),train=self.training)
                     self.condition_token = condition_embeddings
-                    self.condition_token = [self.condition_layer[0](self.condition_token),
-                                            self.condition_layer[1](self.condition_token),
-                                            self.condition_layer[2](self.condition_token)]
+                    self.condition_token = [self.condition_layers[0](self.condition_token),
+                                            self.condition_layers[1](self.condition_token),
+                                            self.condition_layers[2](self.condition_token)]
                     
             else: # decode_n_tokens(kv cache) in inference
                 token_embeddings = self.tok_embeddings(idx)