does this model only supports NVIDA card now?
#365
by
yue32000
- opened
Is possible to run it on amd cards or just cpu?
I'm running it on an m1 Mac with the following code
device = torch.device("mps" if torch.backends.mps.is_available() else "cpu")
print(f'generating on {device}')
pipe = FluxPipeline.from_pretrained("FLUX.1-dev", torch_dtype=torch.bfloat16).to(device)
should work on cpu this way too, just change device to 'cpu' might be quite slow though