manbeast3b commited on
Commit
d31a587
·
verified ·
1 Parent(s): 5cf8ea7

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -43,7 +43,7 @@ def load_pipeline() -> Pipeline:
43
  ckpt_id,
44
  vae=vae,
45
  revision=ckpt_revision,
46
- transformer=model,
47
  # text_encoder_2=text_encoder_2,
48
  torch_dtype=dtype,
49
  ).to(device)
 
43
  ckpt_id,
44
  vae=vae,
45
  revision=ckpt_revision,
46
+ # transformer=model,
47
  # text_encoder_2=text_encoder_2,
48
  torch_dtype=dtype,
49
  ).to(device)