Krebzonide commited on
Commit
99d32d2
·
1 Parent(s): 2438776

adding code to test later

Browse files

I'm in class rn so I can't test it but I have time to write it

Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -3,10 +3,12 @@ from diffusers import StableDiffusionPipeline
3
  import gradio as gr
4
 
5
  model_base = "runwayml/stable-diffusion-v1-5"
 
6
  lora_model_path = "Krebzonide/3a0s-w68r-4qw1-0"
7
 
8
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True)
9
- #pipe.unet.load_attn_procs(lora_model_path)
 
10
  pipe.to("cuda")
11
 
12
  css = """
 
3
  import gradio as gr
4
 
5
  model_base = "runwayml/stable-diffusion-v1-5"
6
+ #model_base = "Krebzonide/LazyMixPlus" test different model checkpoint ----------------------------------------------
7
  lora_model_path = "Krebzonide/3a0s-w68r-4qw1-0"
8
 
9
  pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16, use_safetensors=True)
10
+ #pipe.unet.load_attn_procs(lora_model_path) working, commented to text base model------------------------------------
11
+ #pipe.unet.load_attn_procs(lora_model_path, use_auth_token=True) test accessing a private model----------------------
12
  pipe.to("cuda")
13
 
14
  css = """