Spaces:
Running
on
Zero
Running
on
Zero
dotenv
Browse files- scripts/process_utils.py +1 -1
scripts/process_utils.py
CHANGED
@@ -33,7 +33,7 @@ def ensure_rgb(image):
|
|
33 |
return image
|
34 |
|
35 |
def initialize(_use_local, use_gpu):
|
36 |
-
load_dotenv()
|
37 |
global model, sotai_gen_pipe, refine_gen_pipe, use_local, device, torch_dtype
|
38 |
device = "cuda" if use_gpu and torch.cuda.is_available() else "cpu"
|
39 |
torch_dtype = torch.float16 if device == "cuda" else torch.float32
|
|
|
33 |
return image
|
34 |
|
35 |
def initialize(_use_local, use_gpu):
|
36 |
+
# load_dotenv()
|
37 |
global model, sotai_gen_pipe, refine_gen_pipe, use_local, device, torch_dtype
|
38 |
device = "cuda" if use_gpu and torch.cuda.is_available() else "cpu"
|
39 |
torch_dtype = torch.float16 if device == "cuda" else torch.float32
|