Luiz-Sassi commited on
Commit
67e4235
·
verified ·
1 Parent(s): 9d45241

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -1,7 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
  from PIL import Image
4
- import torch
5
  import spaces
6
 
7
  # 1. Defina o dispositivo padrão (GPU se disponível, senão CPU)
 
1
+ # ==========================================
2
+ # PATCH DE COMPATIBILIDADE (TRANSFORMERS 5.X + RMBG 1.4)
3
+ # Deve ficar no TOPO do arquivo, antes de tudo!
4
+ # ==========================================
5
+ import torch
6
+ _original_getattr = torch.nn.Module.__getattr__
7
+
8
+ def _patched_getattr(self, name):
9
+ if name == "all_tied_weights_keys":
10
+ return {}
11
+ return _original_getattr(self, name)
12
+
13
+ torch.nn.Module.__getattr__ = _patched_getattr
14
+ # ==========================================
15
  import gradio as gr
16
  from transformers import pipeline
17
  from PIL import Image
 
18
  import spaces
19
 
20
  # 1. Defina o dispositivo padrão (GPU se disponível, senão CPU)