Spaces:
Runtime error
Runtime error
Update Feather training runtime image
Browse files- overlay/hydra/model.py +1 -1
overlay/hydra/model.py
CHANGED
|
@@ -508,7 +508,7 @@ class PostSemClawModel(nn.Module):
|
|
| 508 |
|
| 509 |
total_assigned = len(embedding_params) + len(lm_head_params) + len(matrix_params) + len(scalar_params)
|
| 510 |
total_params = len(list(self.parameters()))
|
| 511 |
-
sdr_excluded = len(list(self.sdr_semantic.parameters()))
|
| 512 |
assert total_assigned + sdr_excluded == total_params, (
|
| 513 |
f"Parameter count mismatch: assigned {total_assigned} + sdr_excluded "
|
| 514 |
f"{sdr_excluded} vs total {total_params}"
|
|
|
|
| 508 |
|
| 509 |
total_assigned = len(embedding_params) + len(lm_head_params) + len(matrix_params) + len(scalar_params)
|
| 510 |
total_params = len(list(self.parameters()))
|
| 511 |
+
sdr_excluded = len(list(self.sdr_semantic.parameters())) if self.sdr_semantic is not None else 0
|
| 512 |
assert total_assigned + sdr_excluded == total_params, (
|
| 513 |
f"Parameter count mismatch: assigned {total_assigned} + sdr_excluded "
|
| 514 |
f"{sdr_excluded} vs total {total_params}"
|