sangche commited on
Commit
c0d6e58
·
verified ·
1 Parent(s): d3f2dde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from typing import Sequence, Mapping, Any, Union
5
  import torch
6
  import gradio as gr
7
  from huggingface_hub import hf_hub_download
8
- import spaces
9
  from comfy import model_management
10
 
11
  hf_hub_download(repo_id="Comfy-Org/stable-diffusion-v1-5-archive", filename="v1-5-pruned-emaonly-fp16.safetensors", local_dir="models/checkpoints")
@@ -137,7 +137,7 @@ valid_models = [
137
  #Finally loads the models
138
  model_management.load_models_gpu(valid_models)
139
 
140
- @spaces.GPU(duration=60) # modify the duration for the average it takes for your workflow to run, in seconds
141
  def generate_image(prompt):
142
  import_custom_nodes()
143
  with torch.inference_mode():
 
5
  import torch
6
  import gradio as gr
7
  from huggingface_hub import hf_hub_download
8
+ # import spaces
9
  from comfy import model_management
10
 
11
  hf_hub_download(repo_id="Comfy-Org/stable-diffusion-v1-5-archive", filename="v1-5-pruned-emaonly-fp16.safetensors", local_dir="models/checkpoints")
 
137
  #Finally loads the models
138
  model_management.load_models_gpu(valid_models)
139
 
140
+ # @spaces.GPU(duration=60) # modify the duration for the average it takes for your workflow to run, in seconds
141
  def generate_image(prompt):
142
  import_custom_nodes()
143
  with torch.inference_mode():