Gaze-LLE

Runtime error

soiz1 commited on Feb 7

Commit

7af8b2c

verified ·

1 Parent(s): d8ced7d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ import spaces
 import gradio as gr
 import torch
-torch.backends.cuda.can_use_efficient_attention(False)
 import matplotlib.pyplot as plt
 from PIL import Image, ImageDraw, ImageFont
 import requests

 import gradio as gr
 import torch
+torch.backends.cuda.enable_mem_efficient_attention(False)
 import matplotlib.pyplot as plt
 from PIL import Image, ImageDraw, ImageFont
 import requests