danhtran2mind commited on
Commit
4e08d53
·
verified ·
1 Parent(s): f7a2658

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -48,7 +48,7 @@ import os
48
  ```
49
  ### Load Base Model
50
  ```python
51
- model_id = "danhtran2mind/ghibli-fine-tuned-sd-2.1"
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/ghibli-fine-tuned-sd-2.1-int8")
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
  ```