Update controlnet_bria.py

controlnet_bria.py

@@ -75,7 +75,7 @@ class BriaControlNetModel(ModelMixin, ConfigMixin, PeftAdapterMixin):
         # embedding_dim=self.inner_dim, pooled_projection_dim=pooled_projection_dim
         # )
         self.time_embed = TimestepProjEmbeddings(
-            embedding_dim=self.inner_dim, time_theta=time_theta
+            embedding_dim=self.inner_dim, max_period=time_theta
         )
         
         self.context_embedder = nn.Linear(joint_attention_dim, self.inner_dim)