bleepybloops
commited on
Commit
•
be93480
1
Parent(s):
8b39549
Update README.md
Browse files
README.md
CHANGED
@@ -43,7 +43,11 @@ import random
|
|
43 |
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to("cuda")
|
44 |
|
45 |
lora_weight = 0.8
|
46 |
-
pipe.load_lora_weights(
|
|
|
|
|
|
|
|
|
47 |
pipe.set_adapters('geoguessr_v1', adapter_weights=[lora_weight])
|
48 |
|
49 |
# set params and generate
|
|
|
43 |
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to("cuda")
|
44 |
|
45 |
lora_weight = 0.8
|
46 |
+
pipe.load_lora_weights(
|
47 |
+
'/workspace/geoguessr_v1_000003500.safetensors',
|
48 |
+
weight_name='geoguessr_v1_000003500.safetensors',
|
49 |
+
adapter_name='geoguessr_v1'
|
50 |
+
)
|
51 |
pipe.set_adapters('geoguessr_v1', adapter_weights=[lora_weight])
|
52 |
|
53 |
# set params and generate
|