Hugging Face's logo

Spaces:
nupurkmr9
/
custom-diffusion
Build error

App Files Community
10
Nupur Kumari commited on
Commit
6573419
•
1 Parent(s): 91d41b7

update

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -47,7 +47,7 @@ class InferencePipeline:
47
  pipe = pipe.to(self.device)
48
 
49
  from src import diffuser_training
50
- diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, '<new1>')
51
 
52
  self.pipe = pipe
53
 
 
47
  pipe = pipe.to(self.device)
48
 
49
  from src import diffuser_training
50
+ diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, compress=False)
51
 
52
  self.pipe = pipe
53