Commit
•
54b4891
1
Parent(s):
62e002d
update
Browse files
README.md
CHANGED
|
@@ -47,7 +47,6 @@ from diffusers import StableDiffusionPipeline
|
|
| 47 |
from transformers import CLIPTextModel
|
| 48 |
import torch
|
| 49 |
import os
|
| 50 |
-
os.environ['CUDA_VISIBLE_DEVICES'] = '1'
|
| 51 |
|
| 52 |
pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1").to("cuda")
|
| 53 |
|
|
|
|
| 47 |
from transformers import CLIPTextModel
|
| 48 |
import torch
|
| 49 |
import os
|
|
|
|
| 50 |
|
| 51 |
pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1").to("cuda")
|
| 52 |
|