Hugging Face's logo

jokerbit
/
flux-qa-base

Model card Files Community
jokerbit commited on
Commit
e6fe518
·
verified ·
1 Parent(s): a597918

Fix it

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -33,7 +33,7 @@ def load_pipeline() -> Pipeline:
33
  path,
34
  use_safetensors=False,
35
  local_files_only=True,
36
- torch_dtype=torch.bfloat16).eval()
37
 
38
  pipeline = FluxPipeline.from_pretrained(
39
  CHECKPOINT,
 
33
  path,
34
  use_safetensors=False,
35
  local_files_only=True,
36
+ torch_dtype=torch.bfloat16)
37
 
38
  pipeline = FluxPipeline.from_pretrained(
39
  CHECKPOINT,