Hugging Face's logo

Spaces:
tejani
/
OPENVINO
Runtime error

App Files Community
tejani commited on
Commit
bdbccd8
·
verified ·
1 Parent(s): 356679c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  model_id = "Intel/sd-1.5-square-quantized"
7
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
8
 
9
- # Optional: Move to CPU (default in Spaces) or enable OpenVINO if available
10
  try:
11
  from optimum.intel import OVStableDiffusionPipeline
12
  pipe = OVStableDiffusionPipeline.from_pretrained(model_id, export=False, library_name="diffusers")
 
6
  model_id = "Intel/sd-1.5-square-quantized"
7
  pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
8
 
9
+ # Optional: Try OpenVINO if available
10
  try:
11
  from optimum.intel import OVStableDiffusionPipeline
12
  pipe = OVStableDiffusionPipeline.from_pretrained(model_id, export=False, library_name="diffusers")