Update src/pipeline.py

src/pipeline.py

@@ -47,7 +47,9 @@ def load_pipeline() -> Pipeline:
         text_encoder_2=text_encoder_2,
         torch_dtype=dtype,
         ).to(device)
-    pipeline.transformer = torch.compile(pipeline.transformer, mode="max-autotune")
+    #torch.compile(model: None = None, *, fullgraph: bool = False, dynamic: Optional[bool] = None, backend: Union[str, Callable] = 'inductor', mode: Optional[str] = None, options: Optional[Dict[str, Union[str, int, bool]]] = None, disable: bool = False) → Callable[[Callable[[_InputT], _RetT]], Callable[[_InputT], _RetT]]
+
+    pipeline.transformer = torch.compile(pipeline.transformer, fullgraph=True, mode="max-autotune")
     #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)