Spaces:
Running
on
Zero
Running
on
Zero
Update app_lora1.py
Browse files- app_lora1.py +5 -5
app_lora1.py
CHANGED
|
@@ -138,6 +138,7 @@ def register_scripts(selected_scripts):
|
|
| 138 |
# =========================================================
|
| 139 |
def extract_pipe_lines(script_code: str):
|
| 140 |
lines = script_code.splitlines()
|
|
|
|
| 141 |
pipe_lines = []
|
| 142 |
found = False
|
| 143 |
|
|
@@ -149,6 +150,7 @@ def extract_pipe_lines(script_code: str):
|
|
| 149 |
elif found:
|
| 150 |
if "pipe" in stripped:
|
| 151 |
pipe_lines.append(line)
|
|
|
|
| 152 |
return pipe_lines
|
| 153 |
|
| 154 |
|
|
@@ -164,14 +166,11 @@ def get_pipeline(script_name):
|
|
| 164 |
|
| 165 |
log(f"🔧 Building pipeline from {script_name}")
|
| 166 |
|
| 167 |
-
# --- Log full .py file ---
|
| 168 |
-
log("=== SCRIPT CONTENT START ===")
|
| 169 |
-
log(SCRIPT_CODE[script_name])
|
| 170 |
-
log("=== SCRIPT CONTENT END ===")
|
| 171 |
-
|
| 172 |
namespace = {
|
| 173 |
"__file__": script_name,
|
| 174 |
"__name__": "__main__",
|
|
|
|
|
|
|
| 175 |
"torch": torch,
|
| 176 |
}
|
| 177 |
|
|
@@ -193,6 +192,7 @@ def get_pipeline(script_name):
|
|
| 193 |
return PIPELINES[script_name]
|
| 194 |
|
| 195 |
|
|
|
|
| 196 |
def get_pipeline_fallback(script_name):
|
| 197 |
if script_name in PIPELINES:
|
| 198 |
return PIPELINES[script_name]
|
|
|
|
| 138 |
# =========================================================
|
| 139 |
def extract_pipe_lines(script_code: str):
|
| 140 |
lines = script_code.splitlines()
|
| 141 |
+
print(lines)
|
| 142 |
pipe_lines = []
|
| 143 |
found = False
|
| 144 |
|
|
|
|
| 150 |
elif found:
|
| 151 |
if "pipe" in stripped:
|
| 152 |
pipe_lines.append(line)
|
| 153 |
+
log(f"🔧 Building pipeline from {pipe_lines}")
|
| 154 |
return pipe_lines
|
| 155 |
|
| 156 |
|
|
|
|
| 166 |
|
| 167 |
log(f"🔧 Building pipeline from {script_name}")
|
| 168 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 169 |
namespace = {
|
| 170 |
"__file__": script_name,
|
| 171 |
"__name__": "__main__",
|
| 172 |
+
|
| 173 |
+
# Minimal required globals
|
| 174 |
"torch": torch,
|
| 175 |
}
|
| 176 |
|
|
|
|
| 192 |
return PIPELINES[script_name]
|
| 193 |
|
| 194 |
|
| 195 |
+
|
| 196 |
def get_pipeline_fallback(script_name):
|
| 197 |
if script_name in PIPELINES:
|
| 198 |
return PIPELINES[script_name]
|