Hugging Face's logo

Spaces:
evijit
/
text-to-image-bias
Running on A100

App Files Community
1
Avijit Ghosh commited on
Commit
b9bfe79
•
1 Parent(s): de81f33

add gpu wrapper

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -13,7 +13,6 @@ import os
13
  import spaces
14
 
15
  # Define model initialization functions
16
- @spaces.GPU
17
  def load_model(model_name):
18
  if model_name == "stabilityai/sdxl-turbo":
19
  pipeline = DiffusionPipeline.from_pretrained(
 
13
  import spaces
14
 
15
  # Define model initialization functions
 
16
  def load_model(model_name):
17
  if model_name == "stabilityai/sdxl-turbo":
18
  pipeline = DiffusionPipeline.from_pretrained(