kael558 commited on
Commit
d955083
·
1 Parent(s): 996aa9a

half precision set to false

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. sd.py +1 -1
app.py CHANGED
@@ -77,7 +77,6 @@ def generate_interpolated_images(obj1, obj2):
77
  size2 = json.loads(obj2['size'])
78
  prompt2_c = (torch.from_numpy(arr2.reshape(size2))).float().cuda()
79
 
80
-
81
  images = []
82
  for j in range(1, dist_frames):
83
 
 
77
  size2 = json.loads(obj2['size'])
78
  prompt2_c = (torch.from_numpy(arr2.reshape(size2))).float().cuda()
79
 
 
80
  images = []
81
  for j in range(1, dist_frames):
82
 
sd.py CHANGED
@@ -47,7 +47,7 @@ from ldm.models.diffusion.ddim import DDIMSampler
47
  from ldm.models.diffusion.plms import PLMSSampler
48
 
49
  # 2. Set model download config
50
- def load_model_from_config(config, ckpt, verbose=False, device='cuda', half_precision=True):
51
  map_location = "cuda" if torch.cuda.is_available() else "cpu"
52
  print(f"Loading model from {ckpt}")
53
  pl_sd = torch.load(ckpt, map_location=map_location)
 
47
  from ldm.models.diffusion.plms import PLMSSampler
48
 
49
  # 2. Set model download config
50
+ def load_model_from_config(config, ckpt, verbose=False, device='cuda', half_precision=False):
51
  map_location = "cuda" if torch.cuda.is_available() else "cpu"
52
  print(f"Loading model from {ckpt}")
53
  pl_sd = torch.load(ckpt, map_location=map_location)