agentbot commited on
Commit
7689666
·
verified ·
1 Parent(s): b4bdd09

Initial commit with folder contents

Browse files
Files changed (1) hide show
  1. 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")