Update src/pipeline.py

src/pipeline.py

@@ -45,12 +45,12 @@ def load_pipeline() -> Pipeline:
         ckpt_id,
         vae=vae,
         revision=ckpt_revision,
-        # transformer=model,
-        # text_encoder_2=text_encoder_2,
+        transformer=model,
+        text_encoder_2=text_encoder_2,
         torch_dtype=dtype,
-        )#.to(device)
+        ).to(device)
     pipeline.transformer = torch.compile(pipeline.transformer, mode="max-autotune")
-    quantize_(pipeline.text_encoder, fpx_weight_only(3,2))
+    # quantize_(pipeline.text_encoder, fpx_weight_only(3,2))
     quantize_(pipeline.vae, int8_weight_only())
     for _ in range(3):
         pipeline(prompt="onomancy, aftergo, spirantic, Platyhelmia, modificator, drupaceous, jobbernowl, hereness", width=1024, height=1024, guidance_scale=0.0, num_inference_steps=4, max_sequence_length=256)