TheAIBoi commited on
Commit
d9ef2cf
·
verified ·
1 Parent(s): 7b743a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -12,7 +12,7 @@ model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like
12
  if torch.cuda.is_available():
13
  torch_dtype = torch.float16
14
  else:
15
- torch_dtype = torch.float16
16
 
17
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
18
  pipe = pipe.to(device)
@@ -105,7 +105,7 @@ with gr.Blocks(css=css) as demo:
105
  minimum=256,
106
  maximum=MAX_IMAGE_SIZE,
107
  step=32,
108
- value=1024, # Replace with defaults that work for your model
109
  )
110
 
111
  height = gr.Slider(
@@ -113,7 +113,7 @@ with gr.Blocks(css=css) as demo:
113
  minimum=256,
114
  maximum=MAX_IMAGE_SIZE,
115
  step=32,
116
- value=1024, # Replace with defaults that work for your model
117
  )
118
 
119
  with gr.Row():
@@ -151,4 +151,4 @@ with gr.Blocks(css=css) as demo:
151
  )
152
 
153
  if __name__ == "__main__":
154
- demo.launch()
 
12
  if torch.cuda.is_available():
13
  torch_dtype = torch.float16
14
  else:
15
+ torch_dtype = torch.float32
16
 
17
  pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
18
  pipe = pipe.to(device)
 
105
  minimum=256,
106
  maximum=MAX_IMAGE_SIZE,
107
  step=32,
108
+ value=4096, # Replace with defaults that work for your model
109
  )
110
 
111
  height = gr.Slider(
 
113
  minimum=256,
114
  maximum=MAX_IMAGE_SIZE,
115
  step=32,
116
+ value=4096, # Replace with defaults that work for your model
117
  )
118
 
119
  with gr.Row():
 
151
  )
152
 
153
  if __name__ == "__main__":
154
+ demo.launch(share=True)