Futuretop commited on
Commit
d4af306
·
verified ·
1 Parent(s): 25b1295

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -81,9 +81,10 @@ model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
81
  pipe = DiffusionPipeline.from_pretrained(
82
  LOCAL_TURBOX,
83
  trust_remote_code=True,
 
84
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
85
  )
86
- pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, subfolder="scheduler", trust_remote_code = True, shift=5)
87
  pipe = pipe.to(device)
88
 
89
  MAX_SEED = 2**31 - 1
 
81
  pipe = DiffusionPipeline.from_pretrained(
82
  LOCAL_TURBOX,
83
  trust_remote_code=True,
84
+ local_files_only=True,
85
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
86
  )
87
+ pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, subfolder="scheduler", local_files_only=True, trust_remote_code = True, shift=5)
88
  pipe = pipe.to(device)
89
 
90
  MAX_SEED = 2**31 - 1