Spaces:
Sleeping
Sleeping
peterpeter8585
commited on
Commit
•
966ab30
1
Parent(s):
f2b4626
Update app.py
Browse files
app.py
CHANGED
@@ -7,9 +7,8 @@ import torch
|
|
7 |
import transformers
|
8 |
transformers.utils.move_cache()
|
9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
if torch.cuda.is_available():
|
14 |
torch.cuda.max_memory_allocated(device=device)
|
15 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
|
|
|
7 |
import transformers
|
8 |
transformers.utils.move_cache()
|
9 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
10 |
+
import os
|
11 |
+
os.environ["password"]
|
|
|
12 |
if torch.cuda.is_available():
|
13 |
torch.cuda.max_memory_allocated(device=device)
|
14 |
pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
|