custom-diffusion

Build error

Nupur Kumari commited on Jan 15, 2023

Commit

6573419

1 Parent(s): 91d41b7

update

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -47,7 +47,7 @@ class InferencePipeline:
             pipe = pipe.to(self.device)
         from src import diffuser_training
-        diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, '<new1>')
         self.pipe = pipe

             pipe = pipe.to(self.device)
         from src import diffuser_training
+        diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, compress=False)
         self.pipe = pipe