oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2026-01-23 17:10:41 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix an import

Browse source
This commit is contained in:
oobabooga 2025-04-20 17:51:28 -07:00
parent 8cfd7f976b
commit e243424ba1
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/evaluate.py
View file

@ -2,7 +2,6 @@ import datetime
from pathlib import Path
import pandas as pd
from datasets import load_dataset
from tqdm import tqdm
from modules import shared
@ -39,6 +38,7 @@ def calculate_perplexity(models, input_dataset, stride, _max_length):
'''
import torch
from datasets import load_dataset
from modules.torch_utils import clear_torch_cache