Your Name commited on
Commit
10d457b
1 Parent(s): b2b406b

change path for imports

Browse files
Files changed (1) hide show
  1. 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",