Update README.md
Browse files
README.md
CHANGED
|
@@ -39,7 +39,7 @@ from safetensors.torch import load_model
|
|
| 39 |
from huggingface_hub import hf_hub_download
|
| 40 |
|
| 41 |
MODEL_REGISTRY = {
|
| 42 |
-
'pw_foucast':
|
| 43 |
}
|
| 44 |
ModelClass = MODEL_REGISTRY.get(args.model.lower())
|
| 45 |
|
|
|
|
| 39 |
from huggingface_hub import hf_hub_download
|
| 40 |
|
| 41 |
MODEL_REGISTRY = {
|
| 42 |
+
'pw_foucast': PW_FouCast,
|
| 43 |
}
|
| 44 |
ModelClass = MODEL_REGISTRY.get(args.model.lower())
|
| 45 |
|