Spaces:
Running on Zero

Ruurd commited on
Commit
b3de773
·
verified ·
1 Parent(s): dc427d9

Remove single ]

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -136,7 +136,7 @@ def generate_diffusion_text(input_ids):
136
  input_tensor = torch.tensor([input_ids], dtype=torch.long).to(model.device)
137
  logits = model(input_ids=input_tensor)["logits"]
138
  probs = torch.nn.functional.softmax(logits, dim=-1)[0]
139
- probs = torch.clamp(probs, min=1e-8, max=1.0)]
140
  print("probs", probs)
141
  print("probs min:", probs.min().item(), "max:", probs.max().item(), "sum:", probs.sum().item())
142
  assert torch.all(torch.isfinite(probs)), "Non-finite values in probs!"
 
136
  input_tensor = torch.tensor([input_ids], dtype=torch.long).to(model.device)
137
  logits = model(input_ids=input_tensor)["logits"]
138
  probs = torch.nn.functional.softmax(logits, dim=-1)[0]
139
+ probs = torch.clamp(probs, min=1e-8, max=1.0)
140
  print("probs", probs)
141
  print("probs min:", probs.min().item(), "max:", probs.max().item(), "sum:", probs.sum().item())
142
  assert torch.all(torch.isfinite(probs)), "Non-finite values in probs!"