smartdigitalnetworks commited on
Commit
a1e1b0f
·
verified ·
1 Parent(s): 2f94dc3

Update app_diffusers.py

Browse files
Files changed (1) hide show
  1. app_diffusers.py +3 -2
app_diffusers.py CHANGED
@@ -15,7 +15,7 @@ logging.set_verbosity_error()
15
 
16
  MAX_SEED = np.iinfo(np.int32).max
17
 
18
- device = "cuda" if torch.cuda.is_available() else "cpu"
19
  _dtype = torch.bfloat16
20
  hf_token = os.getenv("HF_TOKEN")
21
 
@@ -118,7 +118,8 @@ Built upon [Ovis-U1](https://huggingface.co/spaces/AIDC-AI/Ovis-U1-3B), Ovis-Ima
118
  examples = examples,
119
  fn = generate,
120
  inputs = [prompt],
121
- outputs = [result]
 
122
  )
123
 
124
  gr.on(
 
15
 
16
  MAX_SEED = np.iinfo(np.int32).max
17
 
18
+ device = "cpu"
19
  _dtype = torch.bfloat16
20
  hf_token = os.getenv("HF_TOKEN")
21
 
 
118
  examples = examples,
119
  fn = generate,
120
  inputs = [prompt],
121
+ outputs = [result],
122
+ cache_examples=True
123
  )
124
 
125
  gr.on(