Initial commit with folder contents
Browse files- src/pipeline.py +0 -2
src/pipeline.py
CHANGED
|
@@ -35,8 +35,6 @@ def load_pipeline() -> Pipeline:
|
|
| 35 |
pipeline.to("cuda")
|
| 36 |
|
| 37 |
try:
|
| 38 |
-
flux_path = path
|
| 39 |
-
flux_path = os.path.join(HF_HUB_CACHE, "models--agentbot--FLUX.1-schnell-int8wo_/snapshots/aa66177be06aba5a88dbe7265255bec48833a936")
|
| 40 |
flux_transformer = FluxTransformer2DModel.from_pretrained(flux_path, torch_dtype=torch.bfloat16, use_safetensors=False)
|
| 41 |
flux_pipeline = FluxPipeline.from_pretrained(ckpt_id, revision=ckpt_revision, transformer=flux_transformer, local_files_only=True, torch_dtype=torch.bfloat16)
|
| 42 |
flux_pipeline.to("cuda")
|
|
|
|
| 35 |
pipeline.to("cuda")
|
| 36 |
|
| 37 |
try:
|
|
|
|
|
|
|
| 38 |
flux_transformer = FluxTransformer2DModel.from_pretrained(flux_path, torch_dtype=torch.bfloat16, use_safetensors=False)
|
| 39 |
flux_pipeline = FluxPipeline.from_pretrained(ckpt_id, revision=ckpt_revision, transformer=flux_transformer, local_files_only=True, torch_dtype=torch.bfloat16)
|
| 40 |
flux_pipeline.to("cuda")
|