Update autoregressive/models/gpt_t2i.py

autoregressive/models/gpt_t2i.py

@@ -429,6 +429,7 @@ class Transformer(nn.Module):
             self.freqs_cis = self.freqs_cis.to(h.device)
         else:
             if cond_idx is not None: # prefill in inference
+                self.control_strength = control_strength
                 token_embeddings = self.cls_embedding(cond_idx, train=self.training)
                 token_embeddings = token_embeddings[:,:self.cls_token_num]
                 if condition is not None: