hsuwill000 commited on
Commit
4265f46
·
verified ·
1 Parent(s): 6904e5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -8,10 +8,6 @@ from huggingface_hub import snapshot_download
8
  import openvino.runtime as ov
9
  from typing import Optional, Dict
10
 
11
- from optimum.intel.openvino.modeling_diffusion import safety_checker
12
-
13
- def sc(self, clip_input, images) : return images, [False for i in images]
14
- safety_checker.StableDiffusionSafetyChecker.forward = sc
15
 
16
  model_id = "hsuwill000/anything-v5-openvino"
17
 
@@ -28,7 +24,8 @@ pipe = OVStableDiffusionPipeline.from_pretrained(
28
  torch_dtype=torch.int8, #快
29
  #torch_dtype=torch.bfloat16, #中
30
  #variant="fp16",
31
- #torch_dtype=torch.IntTensor, #慢
 
32
  use_safetensors=False,
33
  )
34
 
 
8
  import openvino.runtime as ov
9
  from typing import Optional, Dict
10
 
 
 
 
 
11
 
12
  model_id = "hsuwill000/anything-v5-openvino"
13
 
 
24
  torch_dtype=torch.int8, #快
25
  #torch_dtype=torch.bfloat16, #中
26
  #variant="fp16",
27
+ #torch_dtype=torch.IntTensor, #慢,
28
+ safety_checker=None,
29
  use_safetensors=False,
30
  )
31