Spaces:

snap-research
/

weights2weights

Running on Zero

amildravid4292 commited on Jul 21, 2024

Commit

bc4517b

verified ·

1 Parent(s): 2804b00

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -186,7 +186,7 @@ class main():
         self.proj.to(device)
         self.weights.to(device)
-        network = LoRAw2w( self.weights, self.mean, self.std, self.v,
                     self.unet,
                     rank=1,
                     multiplier=1.0,

         self.proj.to(device)
         self.weights.to(device)
+        network = LoRAw2w( self.weights, self.mean, self.std, self.v[:, :1000],
                     self.unet,
                     rank=1,
                     multiplier=1.0,