We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ecb3738 commit f765f96Copy full SHA for f765f96
sharktank/sharktank/models/punet/README.md
@@ -26,6 +26,7 @@ model_dir=$(huggingface-cli download \
26
python -m sharktank.models.punet.tools.import_hf_dataset \
27
--config-json $model_dir/unet/config.json \
28
--output-irpa-file ~/models/punet_fp16.irpa
29
+ --params $model_dir/unet/diffusion_pytorch_model.fp16.safetensors
30
```
31
32
## Running reference model
@@ -37,7 +38,7 @@ python -m sharktank.models.punet.tools.run_diffuser_ref
37
38
## Run punet model
39
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
42
43
44
## Integration Testing
0 commit comments