Spaces:

JVice
/

try-before-you-bias

Running

JVice commited on 13 days ago

Commit

1f43872

•

1 Parent(s): 86c0237

Update model_loading.py

Files changed (1) hide show

model_loading.py CHANGED Viewed

@@ -46,7 +46,6 @@ def import_model(modelID, modelType):
         elif modelType == 'FluxPipeline':
             from diffusers import FluxPipeline
             T2IModel = FluxPipeline.from_pretrained(modelID, torch_dtype=torch.bfloat16)
-            T2IModel.enable_model_cpu_offload()
         else:
             from diffusers import AutoPipelineForText2Image
             T2IModel = AutoPipelineForText2Image.from_pretrained(modelID, torch_dtype=torch.float16)

         elif modelType == 'FluxPipeline':
             from diffusers import FluxPipeline
             T2IModel = FluxPipeline.from_pretrained(modelID, torch_dtype=torch.bfloat16)
         else:
             from diffusers import AutoPipelineForText2Image
             T2IModel = AutoPipelineForText2Image.from_pretrained(modelID, torch_dtype=torch.float16)