Update src/pipeline.py

src/pipeline.py

@@ -81,7 +81,7 @@ def load_pipeline():
         sd = torch.load(p, map_location="cpu", weights_only=True)
         f_sd = {k.strip(pfx): v for k, v in sd.items() if k.strip(pfx) in mod.state_dict() and v.size() == mod.state_dict()[k.strip(pfx)].size()}
         mod.load_state_dict(f_sd, strict=False)
-        #mod.to(dtype=torch.bfloat16)
+        mod.to(dtype=torch.bfloat16)
 
     lsd("ko.pth", vae.encoder, "encoder.")
     lsd("ok.pth", vae.decoder, "decoder.")