Spaces:
Running
on
Zero
Running
on
Zero
heheyas
commited on
Commit
•
21f4df4
1
Parent(s):
eb65de8
update app
Browse files
app.py
CHANGED
|
@@ -80,6 +80,7 @@ def do_sample(
|
|
| 80 |
value_dict["cond_frames"] += 0.05 * torch.randn_like(value_dict["cond_frames"])
|
| 81 |
value_dict["cond_aug"] = 0.05
|
| 82 |
|
|
|
|
| 83 |
with torch.no_grad():
|
| 84 |
with torch.autocast(device):
|
| 85 |
batch, batch_uc = get_batch(
|
|
|
|
| 80 |
value_dict["cond_frames"] += 0.05 * torch.randn_like(value_dict["cond_frames"])
|
| 81 |
value_dict["cond_aug"] = 0.05
|
| 82 |
|
| 83 |
+
print(device)
|
| 84 |
with torch.no_grad():
|
| 85 |
with torch.autocast(device):
|
| 86 |
batch, batch_uc = get_batch(
|