CodedotAl/gpt-code-clippy
1
1
Fork 0
mirror of https://github.com/CodedotAl/gpt-code-clippy.git synced 2024-10-05 20:19:18 +03:00
Code Issues Projects Releases Wiki Activity

updated error in running files

Browse Source
This commit is contained in:
Mrinal18 2021-10-10 00:49:29 -04:00
parent 8e29b6fc51
commit 89e416a82f
4 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
training/run_clm_apps.py
View File

@ -182,7 +182,7 @@ class DataTrainingArguments:
class TrainState(train_state.TrainState):
dropout_rng: jnp.ndarray
dropout_rng: jnp.ndarray = field(default=None)
def replicate(self):
return jax_utils.replicate(self).replace(dropout_rng=shard_prng_key(self.dropout_rng))

2
training/run_clm_flax.py
View File

@ -182,7 +182,7 @@ class DataTrainingArguments:
class TrainState(train_state.TrainState):
dropout_rng: jnp.ndarray
dropout_rng: jnp.ndarray = field(default=None)
def replicate(self):
return jax_utils.replicate(self).replace(dropout_rng=shard_prng_key(self.dropout_rng))

2
training/run_clm_streaming_filter_flax.py
View File

@ -201,7 +201,7 @@ class DataTrainingArguments:
class TrainState(train_state.TrainState):
dropout_rng: jnp.ndarray
dropout_rng: jnp.ndarray = field(default=None)
def replicate(self):
return jax_utils.replicate(self).replace(dropout_rng=shard_prng_key(self.dropout_rng))

2
training/run_clm_streaming_flax.py
View File

@ -200,7 +200,7 @@ class DataTrainingArguments:
class TrainState(train_state.TrainState):
dropout_rng: jnp.ndarray
dropout_rng: jnp.ndarray = field(default=None)
def replicate(self):
return jax_utils.replicate(self).replace(dropout_rng=shard_prng_key(self.dropout_rng))