flax-community
/

gpt2-Cosmos

Model card Files Files and versions

Metrics Training metrics Community

Vivek commited on Jul 16, 2021

Commit

67780a2

·

1 Parent(s): 9e25fd3

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -69,7 +69,7 @@ outputs=model(input_id,att_mask)
 final_output=jnp.argmax(outputs,axis=-1)
-Print(f"the predction of the dataset : {final_output}")
 ```
 ```

 final_output=jnp.argmax(outputs,axis=-1)
+print(f"the predction of the dataset : {final_output}")
 ```
 ```