Hugging Face's logo

Spaces:
MagnaSC
/
FLUX.2-klein-9B
Paused

App Files Community
multimodalart HF Staff commited on
Commit
c31b7c5
·
verified ·
1 Parent(s): e3bed10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -61,7 +61,8 @@ repo_id = "diffusers-internal-dev/dummy-1015-4b" # 4b model
61
  # repo_id = "diffusers-internal-dev/dummy-1015-9b" # 9b model (alternative)
62
 
63
  pipe = Flux2KleinPipeline.from_pretrained(repo_id, torch_dtype=dtype)
64
- pipe.enable_model_cpu_offload()
 
65
 
66
 
67
  def image_to_data_uri(img):
 
61
  # repo_id = "diffusers-internal-dev/dummy-1015-9b" # 9b model (alternative)
62
 
63
  pipe = Flux2KleinPipeline.from_pretrained(repo_id, torch_dtype=dtype)
64
+ pipe.to("cuda")
65
+ #pipe.enable_model_cpu_offload()
66
 
67
 
68
  def image_to_data_uri(img):