recoilme commited on
Commit
7e11826
·
verified ·
1 Parent(s): 4f0170e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -110,7 +110,7 @@ class SimpleDiffusionPipeline(DiffusionPipeline):
110
  #).to(device)
111
 
112
  pipe_id = "AiArtLab/sdxs"
113
- pipe = SdxsPipeline.from_pretrained(
114
  pipe_id,
115
  torch_dtype=dtype,
116
  trust_remote_code=True
 
110
  #).to(device)
111
 
112
  pipe_id = "AiArtLab/sdxs"
113
+ pipe = DiffusionPipeline.from_pretrained(
114
  pipe_id,
115
  torch_dtype=dtype,
116
  trust_remote_code=True