Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -91,7 +91,7 @@ class ModelTrainer:
|
|
91 |
torch.backends.cudnn.allow_tf32 = False
|
92 |
|
93 |
def prepare_initial_datasets(batch_size=8):
|
94 |
-
|
95 |
|
96 |
def process_medqa_batch(examples):
|
97 |
results = []
|
|
|
91 |
torch.backends.cudnn.allow_tf32 = False
|
92 |
|
93 |
def prepare_initial_datasets(batch_size=8):
|
94 |
+
print("Loading datasets with memory-optimized batch processing...")
|
95 |
|
96 |
def process_medqa_batch(examples):
|
97 |
results = []
|