From e243424ba12c1fe030dd5086f67571ad8a6970fa Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 20 Apr 2025 17:51:28 -0700
Subject: [PATCH] Fix an import

---
 modules/evaluate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/evaluate.py b/modules/evaluate.py
index bbf27da1..ba0de378 100644
--- a/modules/evaluate.py
+++ b/modules/evaluate.py
@@ -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