change path for imports
Browse files- handler.py +3 -0
handler.py
CHANGED
@@ -5,6 +5,7 @@ from collections import defaultdict
|
|
5 |
from io import BytesIO
|
6 |
from pprint import pprint
|
7 |
from typing import Any, Dict, List
|
|
|
8 |
|
9 |
import torch
|
10 |
from diffusers import (
|
@@ -35,6 +36,8 @@ if device.type != "cuda":
|
|
35 |
|
36 |
|
37 |
class EndpointHandler:
|
|
|
|
|
38 |
LORA_PATHS = {
|
39 |
"hairdetailer": PREFIX_URL + "lora/hairdetailer.safetensors",
|
40 |
"lora_leica": PREFIX_URL + "lora/lora_leica.safetensors",
|
|
|
5 |
from io import BytesIO
|
6 |
from pprint import pprint
|
7 |
from typing import Any, Dict, List
|
8 |
+
import os
|
9 |
|
10 |
import torch
|
11 |
from diffusers import (
|
|
|
36 |
|
37 |
|
38 |
class EndpointHandler:
|
39 |
+
print(os.getcwd())
|
40 |
+
|
41 |
LORA_PATHS = {
|
42 |
"hairdetailer": PREFIX_URL + "lora/hairdetailer.safetensors",
|
43 |
"lora_leica": PREFIX_URL + "lora/lora_leica.safetensors",
|