Spaces:

HARRY07979
/

SD-v1-5-OpenVINO-DEMO

Runtime error

HARRY07979 commited on Sep 13

Commit

79fee99

verified ·

1 Parent(s): e127256

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import os
 # Thiết lập thiết bị và kiểu dữ liệu
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "segmind/tiny-sd"  # Mô hình Segmind/Tiny-SD
 if torch.cuda.is_available():
     dtype = torch.float16

 # Thiết lập thiết bị và kiểu dữ liệu
 device = "cuda" if torch.cuda.is_available() else "cpu"
+model_repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"  # Mô hình Segmind/Tiny-SD
 if torch.cuda.is_available():
     dtype = torch.float16