FastAPI_lcm_docker

Sleeping

michaelj commited on Nov 23, 2023

Commit

51759f8

1 Parent(s): 1a9d9e0

test

Files changed (1) hide show

model.py CHANGED Viewed

@@ -7,7 +7,7 @@ def get_pipeline():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
     pipe = AutoPipelineForImage2Image.from_pretrained(
         "SimianLuo/LCM_Dreamshaper_v7",
         use_safetensors=True,

     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+    print("文件")
     pipe = AutoPipelineForImage2Image.from_pretrained(
         "SimianLuo/LCM_Dreamshaper_v7",
         use_safetensors=True,