Update modeling_quiet.py
Browse files- modeling_quiet.py +1 -1
modeling_quiet.py
CHANGED
|
@@ -1098,7 +1098,7 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
|
|
| 1098 |
return logits
|
| 1099 |
|
| 1100 |
|
| 1101 |
-
from generate import custom_generate
|
| 1102 |
@torch.no_grad()
|
| 1103 |
def generate(
|
| 1104 |
self,
|
|
|
|
| 1098 |
return logits
|
| 1099 |
|
| 1100 |
|
| 1101 |
+
from .generate import custom_generate
|
| 1102 |
@torch.no_grad()
|
| 1103 |
def generate(
|
| 1104 |
self,
|