slymnyldrm commited on
Commit
681ac98
1 Parent(s): a1b1fbf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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")