Update src/pipeline.py

src/pipeline.py

@@ -129,7 +129,7 @@ def load_pipeline() -> Pipeline:
     # print("no compile")
     # pipeline.to(device)
 
-    pipeline = diffusers.AutoPipelineForText2Image.from_pretrained("Disty0/FLUX.1-dev-qint8", transformer=None, text_encoder_2=None, torch_dtype=torch.bfloat16)
+    pipeline = diffusers.AutoPipelineForText2Image.from_pretrained(ckpt_id, transformer=None, text_encoder_2=None, torch_dtype=torch.bfloat16)
     pipeline.transformer = load_quanto_transformer("Disty0/FLUX.1-dev-qint8")
     pipeline.text_encoder_2 = load_quanto_text_encoder_2("Disty0/FLUX.1-dev-qint8")
     pipeline = pipeline.to(dtype=torch.bfloat16)