Spaces:

gokaygokay
/

Flux-TRELLIS

Running on Zero

gokaygokay commited on 8 days ago

Commit

0e4b750

1 Parent(s): 148741c

spacesgpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -281,7 +281,7 @@ if __name__ == "__main__":
     quantization_config = BitsAndBytesConfig(load_in_4bit=True)
     hf_token = ""
     dtype = torch.bfloat16
-    file_url = "https://huggingface.co/gokaygokay/flux-game/blob/main/gokaygokay_00001_.safetensors"
     single_file_base_model = "camenduru/FLUX.1-dev-diffusers"
     transformer = FluxTransformer2DModel.from_single_file(file_url, subfolder="transformer", torch_dtype=dtype, config=single_file_base_model, quantization_config=quantization_config)
     flux_pipeline = FluxPipeline.from_pretrained(single_file_base_model, transformer=transformer, torch_dtype=dtype, token=huggingface_token, quantization_config=quantization_config)

     quantization_config = BitsAndBytesConfig(load_in_4bit=True)
     hf_token = ""
     dtype = torch.bfloat16
+    file_url = "https://huggingface.co/gokaygokay/flux-game/resolve/main/gokaygokay_00001_.safetensors"
     single_file_base_model = "camenduru/FLUX.1-dev-diffusers"
     transformer = FluxTransformer2DModel.from_single_file(file_url, subfolder="transformer", torch_dtype=dtype, config=single_file_base_model, quantization_config=quantization_config)
     flux_pipeline = FluxPipeline.from_pretrained(single_file_base_model, transformer=transformer, torch_dtype=dtype, token=huggingface_token, quantization_config=quantization_config)