Update app_diffusers.py
Browse files- 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 = "
|
| 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(
|