Spaces:

ajinkyakolhe112
/

mahabharat_fine_tune_flux

Sleeping

ajinkyakolhe112 commited on Dec 8, 2024

Commit

18bbf11

verified ·

1 Parent(s): 42ede63

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
-pipe.load_lora_weights("ajinkyakolhe112/flux-mahabharat-lora", weight_name="my_first_flux_lora_v1.safetensors", adapter_name="mahabharat")
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048

 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
+# pipe.load_lora_weights("ajinkyakolhe112/flux-mahabharat-lora", weight_name="my_first_flux_lora_v1.safetensors", adapter_name="mahabharat")
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048