barakmeiri commited on
Commit
25a833e
·
verified ·
1 Parent(s): 947bcf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -60,9 +60,9 @@ with gr.Blocks(css="style.css") as demo:
60
  scheduler_class = MyEulerAncestralDiscreteScheduler
61
 
62
  print('\n################## 1')
63
- pipe_inversion = SDXLDDIMPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True).to(device)
64
  print('\n################## 2')
65
- pipe_inference = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True).to(device)
66
  print('\n################## 3')
67
  pipe_inference.scheduler = scheduler_class.from_config(pipe_inference.scheduler.config)
68
  pipe_inversion.scheduler = scheduler_class.from_config(pipe_inversion.scheduler.config)
 
60
  scheduler_class = MyEulerAncestralDiscreteScheduler
61
 
62
  print('\n################## 1')
63
+ pipe_inversion = SDXLDDIMPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True).to('cpu')
64
  print('\n################## 2')
65
+ pipe_inference = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True).to('cpu')
66
  print('\n################## 3')
67
  pipe_inference.scheduler = scheduler_class.from_config(pipe_inference.scheduler.config)
68
  pipe_inversion.scheduler = scheduler_class.from_config(pipe_inversion.scheduler.config)