Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ from huggingface_hub import snapshot_download
|
|
5 |
import openvino.runtime as ov
|
6 |
from typing import Optional, Dict
|
7 |
|
8 |
-
model_id = "hsuwill000/
|
9 |
|
10 |
#只有512x512 否則一直重畫
|
11 |
HIGH = 512
|
@@ -33,7 +33,7 @@ pipe = OVStableDiffusionPipeline.from_pretrained(
|
|
33 |
model_id,
|
34 |
compile=False,
|
35 |
ov_config={"CACHE_DIR": ""},
|
36 |
-
torch_dtype=torch.
|
37 |
safety_checker=None,
|
38 |
use_safetensors=False,
|
39 |
)
|
@@ -62,8 +62,8 @@ def infer(prompt, negative_prompt):
|
|
62 |
negative_prompt=negative_prompt,
|
63 |
width=WIDTH,
|
64 |
height=HIGH,
|
65 |
-
guidance_scale=
|
66 |
-
num_inference_steps=
|
67 |
num_images_per_prompt=1,
|
68 |
).images[0]
|
69 |
|
|
|
5 |
import openvino.runtime as ov
|
6 |
from typing import Optional, Dict
|
7 |
|
8 |
+
model_id = "hsuwill000/AngelBreed-v4-openvino-8bit"
|
9 |
|
10 |
#只有512x512 否則一直重畫
|
11 |
HIGH = 512
|
|
|
33 |
model_id,
|
34 |
compile=False,
|
35 |
ov_config={"CACHE_DIR": ""},
|
36 |
+
torch_dtype=torch.uint8, # More standard dtype for speed
|
37 |
safety_checker=None,
|
38 |
use_safetensors=False,
|
39 |
)
|
|
|
62 |
negative_prompt=negative_prompt,
|
63 |
width=WIDTH,
|
64 |
height=HIGH,
|
65 |
+
guidance_scale=7.5
|
66 |
+
num_inference_steps=30,
|
67 |
num_images_per_prompt=1,
|
68 |
).images[0]
|
69 |
|