Spaces:
Sleeping
Sleeping
slymnyldrm
commited on
Commit
•
681ac98
1
Parent(s):
a1b1fbf
Update app.py
Browse files
app.py
CHANGED
@@ -1,9 +1,11 @@
|
|
1 |
import gradio as gr
|
2 |
-
|
3 |
import torch
|
4 |
from torch import autocast
|
5 |
from diffusers import StableDiffusionPipeline, DDIMScheduler
|
6 |
|
|
|
|
|
7 |
model_path = "slymnyldrm/dreambooth_usecase_weights"
|
8 |
|
9 |
pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True, safety_checker=None, torch_dtype=torch.float16).to("cuda")
|
|
|
1 |
import gradio as gr
|
2 |
+
from transformers.utils.hub import move_cache
|
3 |
import torch
|
4 |
from torch import autocast
|
5 |
from diffusers import StableDiffusionPipeline, DDIMScheduler
|
6 |
|
7 |
+
move_cache()
|
8 |
+
|
9 |
model_path = "slymnyldrm/dreambooth_usecase_weights"
|
10 |
|
11 |
pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True, safety_checker=None, torch_dtype=torch.float16).to("cuda")
|