Skip to content

Commit f765f96

Browse files
authored
Fix punet running instructions (#817)
1 parent ecb3738 commit f765f96

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sharktank/sharktank/models/punet/README.md

+2-1
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ model_dir=$(huggingface-cli download \
2626
python -m sharktank.models.punet.tools.import_hf_dataset \
2727
--config-json $model_dir/unet/config.json \
2828
--output-irpa-file ~/models/punet_fp16.irpa
29+
--params $model_dir/unet/diffusion_pytorch_model.fp16.safetensors
2930
```
3031

3132
## Running reference model
@@ -37,7 +38,7 @@ python -m sharktank.models.punet.tools.run_diffuser_ref
3738
## Run punet model
3839

3940
```
40-
python -m sharktank.models.punet.tools.run_punet --irpa-file ~/models/punet_fp16.irpa
41+
python -m sharktank.tools.run_punet --irpa-file ~/models/punet_fp16.irpa
4142
```
4243

4344
## Integration Testing

0 commit comments

Comments
 (0)