rahul7star commited on
Commit
cdf9147
·
verified ·
1 Parent(s): 8b15bc1

Update app_lora1.py

Browse files
Files changed (1) hide show
  1. app_lora1.py +6 -1
app_lora1.py CHANGED
@@ -92,7 +92,7 @@ def register_scripts(selected_scripts):
92
  # GPU-ONLY PIPELINE BUILDER (CRITICAL)
93
  # =========================================================
94
 
95
- @spaces.GPU
96
  def get_pipeline(script_name):
97
  if script_name in PIPELINES:
98
  return PIPELINES[script_name]
@@ -164,6 +164,11 @@ def generate_image(
164
  raise RuntimeError("Pipeline not registered")
165
 
166
  pipe = get_pipeline(pipeline_name)
 
 
 
 
 
167
 
168
  log("=== NEW GENERATION REQUEST ===")
169
  log(f"Pipeline: {pipeline_name}")
 
92
  # GPU-ONLY PIPELINE BUILDER (CRITICAL)
93
  # =========================================================
94
 
95
+
96
  def get_pipeline(script_name):
97
  if script_name in PIPELINES:
98
  return PIPELINES[script_name]
 
164
  raise RuntimeError("Pipeline not registered")
165
 
166
  pipe = get_pipeline(pipeline_name)
167
+
168
+
169
+
170
+ pipe = pipe.to("cuda")
171
+
172
 
173
  log("=== NEW GENERATION REQUEST ===")
174
  log(f"Pipeline: {pipeline_name}")