Update README.md
Browse files
README.md
CHANGED
|
@@ -48,7 +48,7 @@ import os
|
|
| 48 |
```
|
| 49 |
### Load Base Model
|
| 50 |
```python
|
| 51 |
-
model_id = "danhtran2mind/
|
| 52 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 53 |
dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
| 54 |
|
|
@@ -109,6 +109,6 @@ from optimum.intel import OVStableDiffusionPipeline
|
|
| 109 |
```python
|
| 110 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 111 |
|
| 112 |
-
pipe = OVStableDiffusionPipeline.from_pretrained("danhtran2mind/
|
| 113 |
pipe.to(device)
|
| 114 |
```
|
|
|
|
| 48 |
```
|
| 49 |
### Load Base Model
|
| 50 |
```python
|
| 51 |
+
model_id = "danhtran2mind/Ghibli-Stable-Diffusion-2.1-Base-finetuning"
|
| 52 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 53 |
dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
| 54 |
|
|
|
|
| 109 |
```python
|
| 110 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 111 |
|
| 112 |
+
pipe = OVStableDiffusionPipeline.from_pretrained("danhtran2mind/Ghibli-Stable-Diffusion-2.1-Base-finetuning-INT8")
|
| 113 |
pipe.to(device)
|
| 114 |
```
|