radames commited on
Commit
55e36ea
1 Parent(s): c5db356

print info

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -7,6 +7,13 @@ from user_queue import user_data
7
  from util import get_pipeline_class
8
 
9
 
 
 
 
 
 
 
 
10
  app = FastAPI()
11
 
12
  pipeline_class = get_pipeline_class(args.pipeline)
 
7
  from util import get_pipeline_class
8
 
9
 
10
+ print("DEVICE:", device)
11
+ print("TORCH_DTYPE:", torch_dtype)
12
+ print("PIPELINE:", args.pipeline)
13
+ print("SAFETY_CHECKER:", args.safety_checker)
14
+ print("TORCH_COMPILE:", args.torch_compile)
15
+
16
+
17
  app = FastAPI()
18
 
19
  pipeline_class = get_pipeline_class(args.pipeline)