Spaces:
Running on Zero
Running on Zero
fix: use eager transformer instead of incompatible AOTI blocks (#7)
Browse files- fix: use eager transformer instead of incompatible AOTI blocks (499a1c7920d2b60a875f85f2833d3bbb8ed8ad43)
app.py
CHANGED
|
@@ -47,8 +47,8 @@ pipe.set_adapters(["angles"], adapter_weights=[1.])
|
|
| 47 |
pipe.fuse_lora(adapter_names=["angles"], lora_scale=1.25)
|
| 48 |
pipe.unload_lora_weights()
|
| 49 |
|
| 50 |
-
#
|
| 51 |
-
|
| 52 |
|
| 53 |
MAX_SEED = np.iinfo(np.int32).max
|
| 54 |
|
|
|
|
| 47 |
pipe.fuse_lora(adapter_names=["angles"], lora_scale=1.25)
|
| 48 |
pipe.unload_lora_weights()
|
| 49 |
|
| 50 |
+
# Keep the transformer in eager mode: the legacy FA3 AOTI blocks fail on
|
| 51 |
+
# current ZeroGPU hardware with "no kernel image is available for execution".
|
| 52 |
|
| 53 |
MAX_SEED = np.iinfo(np.int32).max
|
| 54 |
|