vampnet

Running on T4

Hugo Flores Garcia commited on Jul 11, 2023

Commit

7053a14

•

1 Parent(s): b1581a7

temp float

Files changed (1) hide show

app.py CHANGED Viewed

@@ -115,7 +115,7 @@ def _vamp(data, return_mask=False):
         z,
         mask=mask,
         sampling_steps=data[num_steps],
-        temperature=data[temp]*10,
         return_mask=True,
         typical_filtering=data[typical_filtering],
         typical_mass=data[typical_mass],

         z,
         mask=mask,
         sampling_steps=data[num_steps],
+        temperature=float(data[temp]*10),
         return_mask=True,
         typical_filtering=data[typical_filtering],
         typical_mass=data[typical_mass],