Update worker.py
Browse files
worker.py
CHANGED
|
@@ -355,6 +355,8 @@ def is_parsable_json(command):
|
|
| 355 |
|
| 356 |
@torch.inference_mode()
|
| 357 |
def generate(command):
|
|
|
|
|
|
|
| 358 |
print(command)
|
| 359 |
if is_parsable_json(command):
|
| 360 |
values = json.loads(command)
|
|
|
|
| 355 |
|
| 356 |
@torch.inference_mode()
|
| 357 |
def generate(command):
|
| 358 |
+
print(command)
|
| 359 |
+
command = command["input"]
|
| 360 |
print(command)
|
| 361 |
if is_parsable_json(command):
|
| 362 |
values = json.loads(command)
|