Update README.md
Browse files
README.md
CHANGED
|
@@ -33,6 +33,8 @@ You can run the smashed model on a A100 GPU by:
|
|
| 33 |
3. Loading the model
|
| 34 |
4. Running the model.
|
| 35 |
|
|
|
|
|
|
|
| 36 |
You can achieve this by running the following code:
|
| 37 |
|
| 38 |
```python
|
|
|
|
| 33 |
3. Loading the model
|
| 34 |
4. Running the model.
|
| 35 |
|
| 36 |
+
The model is smashed for 512x512 image size. Contact us if you want a version with dynamic size support.
|
| 37 |
+
|
| 38 |
You can achieve this by running the following code:
|
| 39 |
|
| 40 |
```python
|