SimpleStories
/

SimpleStories-125M

Text Generation

model_hub_mixin

pytorch_model_hub_mixin

Model card Files Files and versions

lennart-finke commited on Dec 9, 2024

Commit

7095bf6

·

verified ·

1 Parent(s): 6c72502

Fixed readme

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -10,11 +10,12 @@ tags:
 datasets:
 - lennart-finke/SimpleStories
 ---
-For loading this model from within [https://github.com/danbraunai/simple_stories_train](), you can run:
 ```python
 from typing import Any
 import torch.nn as nn
 from huggingface_hub import PyTorchModelHubMixin

 datasets:
 - lennart-finke/SimpleStories
 ---
+For loading this model from within the [https://github.com/danbraunai/simple_stories_train](https://github.com/danbraunai/simple_stories_train), you can run:
 ```python
 from typing import Any
+import torch
 import torch.nn as nn
 from huggingface_hub import PyTorchModelHubMixin