Spaces:

snap-research
/

KontinuousKontext

Running on Zero

RishubhPar commited on Oct 30

Commit

832bd45

verified ·

1 Parent(s): 1948ca5

changed dtype

Files changed (1) hide show

app.py CHANGED Viewed

@@ -159,7 +159,7 @@ transformer = FluxTransformer2DModelwithSliderConditioning.from_pretrained(
     subfolder="transformer",
     token=token,
     trust_remote_code=True,
-    torch_dtype=dtype,
     # low_cpu_mem_usage=True,
     # device_map="balanced_low_0",
     # offload_folder=offload_dir,

     subfolder="transformer",
     token=token,
     trust_remote_code=True,
+    # torch_dtype=dtype,
     # low_cpu_mem_usage=True,
     # device_map="balanced_low_0",
     # offload_folder=offload_dir,