Update app.py

app.py

@@ -82,6 +82,19 @@ PRESETS = {
 #     token=HF_TOKEN,
 # )
 
+
+if TEXT_ENCODER_ID:
+    text_encoder = AutoModel.from_pretrained(
+        TEXT_ENCODER_ID,
+        torch_dtype=torch.bfloat16,
+        token=HF_TOKEN,
+        low_cpu_mem_usage=True,
+    )
+    print(f"[model] using alternate text encoder: {TEXT_ENCODER_ID}", flush=True)
+else:
+    text_encoder = None
+
+    
 pipe = Ideogram4Pipeline.from_pretrained(
     BASE_REPO,
     text_encoder=text_encoder,