Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -15,13 +15,6 @@ train_batch_size = 4
|
|
15 |
gradient_accumulation_steps = 2
|
16 |
output_dir = "flux-anime"
|
17 |
|
18 |
-
# Inisialisasi Hugging Face Hub
|
19 |
-
if whoami().get("name", None) is None:
|
20 |
-
raise ValueError("Login ke Hugging Face Hub terlebih dahulu.")
|
21 |
-
username = whoami()["name"]
|
22 |
-
repo_id = f"{username}/{output_dir}"
|
23 |
-
repo = Repository(output_dir, clone_from=repo_id)
|
24 |
-
|
25 |
# Muat model dan scheduler
|
26 |
pipeline = DiffusionPipeline.from_pretrained(pretrained_model_name_or_path, torch_dtype=torch.float16)
|
27 |
pipeline.scheduler = DDPMScheduler.from_config(pipeline.scheduler.config)
|
|
|
15 |
gradient_accumulation_steps = 2
|
16 |
output_dir = "flux-anime"
|
17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
# Muat model dan scheduler
|
19 |
pipeline = DiffusionPipeline.from_pretrained(pretrained_model_name_or_path, torch_dtype=torch.float16)
|
20 |
pipeline.scheduler = DDPMScheduler.from_config(pipeline.scheduler.config)
|