ironjr commited on
Commit
115ab5a
1 Parent(s): d057fb7

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +4 -4
model.py CHANGED
@@ -196,11 +196,11 @@ class StreamMultiDiffusion(nn.Module):
196
 
197
  print(f'[INFO] Model is loaded!')
198
 
199
- self.reset_seed(self.generator, seed)
200
- self.reset_latent()
201
- self.prepare()
202
 
203
- print(f'[INFO] Parameters prepared!')
204
 
205
  self.ready_checklist['initialized'] = True
206
 
 
196
 
197
  print(f'[INFO] Model is loaded!')
198
 
199
+ # self.reset_seed(self.generator, seed)
200
+ # self.reset_latent()
201
+ # self.prepare()
202
 
203
+ # print(f'[INFO] Parameters prepared!')
204
 
205
  self.ready_checklist['initialized'] = True
206