Spaces:
Running
on
Zero
Running
on
Zero
Remove single ]
Browse files
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!"
|