fix #180

kingoflolz · Jan 28, 2022 · 0a75ca9 · 0a75ca9
1 parent 7b3507c
commit 0a75ca9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/to_hf_weights.py b/to_hf_weights.py
@@ -309,7 +309,7 @@ def unshard_leave(
 
     # stack leave shards into single np.ndarray
     x = np.stack(leave_shards)
-    assert isinstance(x, jnp.ndarray)
+    # assert isinstance(x, jnp.ndarray)
 
     # As far as i can tell, this just re labels the dtype of arrays
     # labeled with "V2" dtype. In theory, V2 was just an alias for bfloat16