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",
|