Spaces:

rahul7star
/

Zimg-debug

Running on Zero

rahul7star commited on 10 days ago

Commit

6a66448

verified ·

1 Parent(s): 93a4a97

Update app_lora1.py

Files changed (1) hide show

app_lora1.py CHANGED Viewed

@@ -44,8 +44,7 @@ try:
         torch_dtype=DTYPE,
     )
-    if hasattr(pipe, "transformer") and hasattr(pipe.transformer, "set_attention_backend"):
-        pipe.transformer.set_attention_backend("_flash_3")
     pipe.to(DEVICE)
     log("✅ Pipeline built successfully")

         torch_dtype=DTYPE,
     )
     pipe.to(DEVICE)
     log("✅ Pipeline built successfully")