Cun-Duck commited on
Commit
bc5e1f8
·
verified ·
1 Parent(s): 7aec5a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -7
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)