amildravid4292 commited on
Commit
5a0f91a
·
verified ·
1 Parent(s): bc4517b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -186,7 +186,7 @@ class main():
186
  self.proj.to(device)
187
  self.weights.to(device)
188
 
189
- network = LoRAw2w( self.weights, self.mean, self.std, self.v[:, :1000],
190
  self.unet,
191
  rank=1,
192
  multiplier=1.0,
 
186
  self.proj.to(device)
187
  self.weights.to(device)
188
 
189
+ network = LoRAw2w( self.weights.bfloat16(), self.mean.bfloat16(), self.std.bfloat16(), self.v[:, :1000].bfloat16(),
190
  self.unet,
191
  rank=1,
192
  multiplier=1.0,