Hugging Face's logo

Spaces:
tonyassi
/
lucy-text-to-image
Running on Zero

App Files Community
tonyassi commited on
Commit
7fbef3a
·
1 Parent(s): 5d48970

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -3,12 +3,14 @@ import torch
3
  from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
4
 
5
 
6
-
7
  model_base = "runwayml/stable-diffusion-v1-5"
8
 
9
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True, safety_checker=None,)
10
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
11
 
 
 
 
12
 
13
 
14
  def greet(description,color,features,occasion,type):
 
3
  from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
4
 
5
 
 
6
  model_base = "runwayml/stable-diffusion-v1-5"
7
 
8
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True, safety_checker=None,)
9
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
10
 
11
+ lora_model_path = "./loralucy3/checkpoint-95000"
12
+ pipe.unet.load_attn_procs(lora_model_path)
13
+ pipe.to("cuda")
14
 
15
 
16
  def greet(description,color,features,occasion,type):