Update README.md
#10
by
hypothetical
- opened
README.md
CHANGED
|
@@ -35,7 +35,7 @@ __Goals of Elastic Models:__
|
|
| 35 |
## Inference
|
| 36 |
|
| 37 |
Currently, our demo model only supports 1024x1024 outputs without batching. This will be updated in the near future.
|
| 38 |
-
To infer our models, you just need to replace `
|
| 39 |
|
| 40 |
```python
|
| 41 |
import torch
|
|
|
|
| 35 |
## Inference
|
| 36 |
|
| 37 |
Currently, our demo model only supports 1024x1024 outputs without batching. This will be updated in the near future.
|
| 38 |
+
To infer our models, you just need to replace `diffusers` import with `elastic_models.diffusers`:
|
| 39 |
|
| 40 |
```python
|
| 41 |
import torch
|