Update app.py

app.py

@@ -21,7 +21,19 @@ if torch.cuda.is_available():
         add_watermarker=False
     )
     #pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
-    pipe.tokenizer.model_max_length=512
+    # 设置 tokenizer 的最大长度
+    max_token_length = 512
+    pipe.tokenizer.model_max_length = max_token_length
+    
+    # 调整文本编码器的配置
+    pipe.text_encoder.config.max_position_embeddings = max_token_length
+    
+    # 如果需要，重新初始化位置嵌入
+    import torch.nn as nn
+    old_emb = pipe.text_encoder.text_model.embeddings.position_embedding.weight.data
+    new_emb = nn.Parameter(torch.zeros(max_token_length, old_emb.shape[1]))
+    new_emb.data[:old_emb.shape[0], :] = old_emb
+    pipe.text_encoder.text_model.embeddings.position_embedding.weight = new_emb
     pipe.to("cuda")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int: