AustingDong
commited on
Commit
·
52aa2d5
1
Parent(s):
73c356e
Update visualization.py
Browse files- demo/visualization.py +0 -1
demo/visualization.py
CHANGED
|
@@ -8,7 +8,6 @@ from PIL import Image
|
|
| 8 |
from torch import nn
|
| 9 |
import spaces
|
| 10 |
from demo.modify_llama import *
|
| 11 |
-
from demo.modified_attn import ModifiedLlamaAttention
|
| 12 |
|
| 13 |
class Visualization:
|
| 14 |
def __init__(self, model, register=True):
|
|
|
|
| 8 |
from torch import nn
|
| 9 |
import spaces
|
| 10 |
from demo.modify_llama import *
|
|
|
|
| 11 |
|
| 12 |
class Visualization:
|
| 13 |
def __init__(self, model, register=True):
|