Views
No views yet
1import torch
2from diffusers import FluxPipeline
3import time
4import random
5
6# initialize pipeline and lora
7pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to("cuda")
8
9lora_weight = 0.8
10pipe.load_lora_weights(
11 '/workspace/geoguessr_v1_000003500.safetensors',
12 adapter_name='geoguessr_v1'
13)
14pipe.set_adapters('geoguessr_v1', adapter_weights=[lora_weight])
15
16# set params and generate
17seed = -1
18seed = seed if seed != -1 else random.randint(0, 2**32)
19print(seed)
20
21prompt = "sweden, snow"
22out = pipe(
23 prompt=prompt,
24 guidance_scale=4,
25 height=624,
26 width=960,
27 num_inference_steps=40,
28 generator=torch.Generator("cuda").manual_seed(seed),
29).images[0]
30
31# save and display output
32filename=f"{time.time()}.png"
33out.save(filename)
34
35from IPython.display import Image, display
36display(Image(filename=filename))
37