Spaces:
Sleeping
Sleeping
transformers cache
Browse files
app.py
CHANGED
@@ -8,6 +8,8 @@ from huggingface_hub import login
|
|
8 |
# import spaces #[uncomment to use ZeroGPU]
|
9 |
from diffusers import DiffusionPipeline
|
10 |
import torch
|
|
|
|
|
11 |
|
12 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
13 |
model_repo_id = "stabilityai/stable-diffusion-3.5-large" # Replace to the model you would like to use
|
|
|
8 |
# import spaces #[uncomment to use ZeroGPU]
|
9 |
from diffusers import DiffusionPipeline
|
10 |
import torch
|
11 |
+
import transformers
|
12 |
+
transformers.utils.move_cache()
|
13 |
|
14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
15 |
model_repo_id = "stabilityai/stable-diffusion-3.5-large" # Replace to the model you would like to use
|