Spaces:

ekt1701
/

model_testing

Sleeping

ekt1701 commited on May 16

Commit

1f9b0ed

verified ·

1 Parent(s): 90b14bc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ MAX_IMAGE_SIZE = 2048
 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
     transformer = FluxTransformer2DModel.from_single_file(
-        "https://huggingface.co/nathann55/Project0_Realism_FP16/resolve/main/project0.safetensors",
         torch_dtype=torch.bfloat16
     )
     pipe = FluxPipeline.from_pretrained(

 # Ensure model and scheduler are initialized in GPU-enabled function
 if torch.cuda.is_available():
     transformer = FluxTransformer2DModel.from_single_file(
+        "https://huggingface.co/nathann55/Project0_Realism_FP16/blob/main/project0.safetensors",
         torch_dtype=torch.bfloat16
     )
     pipe = FluxPipeline.from_pretrained(