Update README.md
Browse files
README.md
CHANGED
|
@@ -133,8 +133,6 @@ pipe.transformer.eval()
|
|
| 133 |
pipe.vae.eval()
|
| 134 |
pipe.text_encoder.eval()
|
| 135 |
pipe.pe.eval()
|
| 136 |
-
# 如果是消费级显卡,例如 Nvidia 3090
|
| 137 |
-
# pipe.enable_model_cpu_offload()
|
| 138 |
|
| 139 |
# 设置随机种子
|
| 140 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
|
|
|
| 133 |
pipe.vae.eval()
|
| 134 |
pipe.text_encoder.eval()
|
| 135 |
pipe.pe.eval()
|
|
|
|
|
|
|
| 136 |
|
| 137 |
# 设置随机种子
|
| 138 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|