update replicate predict.py
This commit is contained in:
2
README.md
vendored
2
README.md
vendored
@@ -6,7 +6,7 @@
|
||||
|
||||
[](https://discord.gg/xBPBXfcFHd)
|
||||
|
||||
This is a minimal implementation of Boris Dayma's [DALL·E Mini](https://github.com/borisdayma/dalle-mini) in PyTorch. It has been stripped to the bare essentials necessary for doing inference. The only third party dependencies are numpy, requests, pillow and torch.
|
||||
This is a fast, minimal implementation of Boris Dayma's [DALL·E Mini](https://github.com/borisdayma/dalle-mini) in PyTorch. It has been stripped to the bare essentials necessary for doing inference. The only third party dependencies are numpy, requests, pillow and torch.
|
||||
|
||||
To generate a 3x3 grid of DALL·E Mega images it takes
|
||||
- **35 seconds** with a P100 in Colab
|
||||
|
@@ -20,7 +20,7 @@ class Predictor(BasePredictor):
|
||||
grid_size: int = Input(
|
||||
description='Grid Size',
|
||||
ge=1,
|
||||
le=3,
|
||||
le=4,
|
||||
default=3
|
||||
)
|
||||
) -> Path:
|
||||
|
Reference in New Issue
Block a user