Update worker_runpod.py
Browse files- worker_runpod.py +1 -5
worker_runpod.py
CHANGED
|
@@ -1,8 +1,4 @@
|
|
| 1 |
-
import os,
|
| 2 |
-
|
| 3 |
-
directory_path = "/content/DiffBIR"
|
| 4 |
-
os.chdir(directory_path)
|
| 5 |
-
sys.path.append(directory_path)
|
| 6 |
|
| 7 |
import torch
|
| 8 |
from accelerate.utils import set_seed
|
|
|
|
| 1 |
+
import os, shutil, json, requests, random, runpod
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
import torch
|
| 4 |
from accelerate.utils import set_seed
|