Spaces:

rafaaa2105
/

diffusers-collection

Sleeping

rafaaa2105 commited on May 15

Commit

809d6c5

•

1 Parent(s): 49f05ba

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import torch
 from tqdm import tqdm
 from diffusers.models import AutoencoderKL
-from diffusers import StableDiffusionPipeline
 import gradio as gr
 import requests
 import spaces
@@ -92,7 +92,7 @@ def load_model(model, lora="", use_lora=False, progress=gr.Progress(track_tqdm=T
             torch_dtype=torch.float16,
         )
-        pipeline = StableDiffusionPipeline.from_ckpt(
             model,
             vae=vae,
             torch_dtype=torch.float16,

 import torch
 from tqdm import tqdm
 from diffusers.models import AutoencoderKL
+from diffusers import StableDiffusionXLPipeline
 import gradio as gr
 import requests
 import spaces
             torch_dtype=torch.float16,
         )
+        pipeline = StableDiffusionPipelineXL.from_single_file(
             model,
             vae=vae,
             torch_dtype=torch.float16,