FastAPI_lcm_docker

Running

michaelj commited on Nov 23, 2023

Commit

e4b8c77

•

1 Parent(s): 1e2deb3

add

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
-    print("重新下载")
     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,