Update app.py
This commit is contained in:
2
app.py
2
app.py
@@ -70,7 +70,7 @@ def upload_and_prepare_dataset(files, dataset_name, trigger_word):
|
||||
return f"✅ ready: {image_count} images in {dataset_name}", str(dataset_path), dataset_name
|
||||
|
||||
# request 10 mins gpu for training
|
||||
@spaces.GPU(duration=600)
|
||||
@spaces.GPU(duration=200)
|
||||
def train_lora(
|
||||
dataset_path,
|
||||
project_name,
|
||||
|
||||
Reference in New Issue
Block a user